|
..
|
|
device
|
|
eager
|
|
gpu
|
|
next_pluggable_device
|
|
pluggable_device
|
|
BUILD
|
|
accumulate_n_optimizer.cc
|
|
all_to_all.cc
|
|
all_to_all.h
|
|
all_to_all_test.cc
|
|
allocator_retry.h
|
|
arg_ret_placement.cc
|
|
arg_ret_placement.h
|
|
arg_ret_placement_test.cc
|
|
base_collective_executor.cc
|
|
base_collective_executor.h
|
|
bfc_allocator.h
|
|
buf_rendezvous.cc
|
|
buf_rendezvous.h
|
|
buf_rendezvous_test.cc
|
|
build_graph_options.cc
|
|
build_graph_options.h
|
|
collective_executor_mgr.cc
|
|
collective_executor_mgr.h
|
|
collective_executor_mgr_test.cc
|
|
collective_param_resolver_local.cc
|
|
collective_param_resolver_local.h
|
|
collective_param_resolver_local_test.cc
|
|
collective_rma_local.cc
|
|
collective_rma_local.h
|
|
collective_rma_local_test.cc
|
|
collective_test_util.cc
|
|
collective_test_util.h
|
|
collective_util.cc
|
|
collective_util.h
|
|
colocation_graph.cc
|
|
colocation_graph.h
|
|
composite_device.cc
|
|
composite_device.h
|
|
composite_device_test.cc
|
|
constant_folding.cc
|
|
constant_folding.h
|
|
constant_folding_test.cc
|
|
copy_tensor.cc
|
|
copy_tensor.h
|
|
cost_constants.h
|
|
cost_measurement.h
|
|
cost_measurement_registry.cc
|
|
cost_measurement_registry.h
|
|
cost_measurement_registry_test.cc
|
|
cost_util.cc
|
|
cost_util.h
|
|
cost_util_test.cc
|
|
costmodel_manager.cc
|
|
costmodel_manager.h
|
|
debugger_state_interface.cc
|
|
debugger_state_interface.h
|
|
device.h
|
|
device_factory.h
|
|
device_mgr.cc
|
|
device_mgr.h
|
|
device_mgr_test.cc
|
|
device_propagation.cc
|
|
device_propagation.h
|
|
device_propagation_test.cc
|
|
device_resolver_local.cc
|
|
device_resolver_local.h
|
|
device_resolver_local_test.cc
|
|
device_set.cc
|
|
device_set.h
|
|
device_set_test.cc
|
|
direct_session.cc
|
|
direct_session.h
|
|
direct_session_test.cc
|
|
direct_session_with_tracking_alloc_test.cc
|
|
dma_helper.h
|
|
dynamic_device_mgr.cc
|
|
dynamic_device_mgr_test.cc
|
|
entry.h
|
|
eval_const_tensor.cc
|
|
eval_const_tensor.h
|
|
eval_const_tensor_test.cc
|
|
executor.cc
|
|
executor.h
|
|
executor_factory.cc
|
|
executor_factory.h
|
|
executor_test.cc
|
|
function.cc
|
|
function.h
|
|
function_body.cc
|
|
function_body.h
|
|
function_def_utils.cc
|
|
function_def_utils.h
|
|
function_optimization_registration_test.cc
|
|
function_optimization_registry.cc
|
|
function_optimization_registry.h
|
|
function_optimization_registry_no_pass_test.cc
|
|
function_optimization_registry_pass_failure_test.cc
|
|
function_optimization_registry_test.cc
|
|
function_test.cc
|
|
function_testlib.cc
|
|
function_testlib.h
|
|
function_threadpool_test.cc
|
|
function_utils.cc
|
|
function_utils.h
|
|
gpu_device_context.h
|
|
gradients.cc
|
|
gradients.h
|
|
graph_constructor.cc
|
|
graph_constructor.h
|
|
graph_constructor_fuzz.cc
|
|
graph_constructor_test.cc
|
|
graph_def_builder_util.cc
|
|
graph_def_builder_util.h
|
|
graph_execution_state.cc
|
|
graph_execution_state.h
|
|
graph_optimizer.cc
|
|
graph_optimizer.h
|
|
graph_runner.cc
|
|
graph_runner.h
|
|
graph_runner_test.cc
|
|
graph_view.cc
|
|
graph_view.h
|
|
hierarchical_tree_broadcaster.cc
|
|
hierarchical_tree_broadcaster.h
|
|
hierarchical_tree_broadcaster_test.cc
|
|
immutable_executor_state.cc
|
|
immutable_executor_state.h
|
|
inline_function_utils.cc
|
|
inline_function_utils.h
|
|
inline_function_utils_test.cc
|
|
input_colocation_exemption_registry.cc
|
|
input_colocation_exemption_registry.h
|
|
input_colocation_exemption_registry_test.cc
|
|
inspecting_placer.cc
|
|
inspecting_placer.h
|
|
int32_fulltype.cc
|
|
int32_fulltype.h
|
|
int32_fulltype_test.cc
|
|
isolate_placer_inspection_required_ops_pass.cc
|
|
isolate_placer_inspection_required_ops_pass.h
|
|
isolate_placer_inspection_required_ops_pass_test.cc
|
|
kernel_benchmark_testlib.cc
|
|
kernel_benchmark_testlib.h
|
|
layout_pass_util.cc
|
|
layout_pass_util.h
|
|
local_device.cc
|
|
local_device.h
|
|
local_executor_params.h
|
|
local_session_selection.cc
|
|
local_session_selection.h
|
|
lower_case_op.cc
|
|
lower_case_op.h
|
|
lower_case_op_test.cc
|
|
lower_function_call_inline_policy.cc
|
|
lower_function_call_inline_policy.h
|
|
lower_function_call_op.cc
|
|
lower_function_call_op.h
|
|
lower_function_call_op_test.cc
|
|
lower_functional_ops.cc
|
|
lower_functional_ops.h
|
|
lower_functional_ops_test.cc
|
|
lower_if_op.cc
|
|
lower_if_op.h
|
|
lower_if_op_test.cc
|
|
lower_while_op.cc
|
|
lower_while_op.h
|
|
lower_while_op_test.cc
|
|
memory_types.cc
|
|
memory_types.h
|
|
memory_types_test.cc
|
|
mkl_cpu_allocator.cc
|
|
mkl_cpu_allocator.h
|
|
mkl_cpu_allocator_test.cc
|
|
mkl_layout_pass.cc
|
|
mkl_layout_pass.h
|
|
mkl_threadpool_device_test.cc
|
|
no_op_cost_measurement.cc
|
|
no_op_cost_measurement.h
|
|
no_op_cost_measurement_test.cc
|
|
node_file_writer.cc
|
|
node_file_writer.h
|
|
null_request_cost_accessor.cc
|
|
null_request_cost_accessor.h
|
|
null_request_cost_accessor_test.cc
|
|
optimization_registry.cc
|
|
optimization_registry.h
|
|
optimization_registry_test.cc
|
|
optimize_cross_host_control_deps.cc
|
|
optimize_cross_host_control_deps.h
|
|
optimize_cross_host_control_deps_test.cc
|
|
optimize_function_graph_utils.cc
|
|
optimize_function_graph_utils.h
|
|
optimize_function_graph_utils_test.cc
|
|
optimized_function_graph_info.cc
|
|
optimized_function_graph_info.h
|
|
optimized_function_graph_info_test.cc
|
|
parallel_concat_optimizer.cc
|
|
partitioning_utils.cc
|
|
partitioning_utils.h
|
|
partitioning_utils_test.cc
|
|
pending_counts.h
|
|
pending_counts_test.cc
|
|
permuter.cc
|
|
permuter.h
|
|
permuter_test.cc
|
|
placer.cc
|
|
placer.h
|
|
placer_inspection_required_ops_utils.cc
|
|
placer_inspection_required_ops_utils.h
|
|
placer_inspection_required_ops_utils_test.cc
|
|
placer_test.cc
|
|
pool_allocator.cc
|
|
pool_allocator.h
|
|
process_function_library_runtime.cc
|
|
process_function_library_runtime.h
|
|
process_function_library_runtime_test.cc
|
|
process_state.cc
|
|
process_state.h
|
|
process_util.cc
|
|
process_util.h
|
|
process_util_test.cc
|
|
profile_handler.h
|
|
propagator_debug_utils.cc
|
|
propagator_debug_utils.h
|
|
propagator_state.cc
|
|
propagator_state.h
|
|
quantize_training.cc
|
|
quantize_training.h
|
|
quantize_training_test.cc
|
|
renamed_device.cc
|
|
renamed_device.h
|
|
rendezvous_mgr.cc
|
|
rendezvous_mgr.h
|
|
rendezvous_util.cc
|
|
rendezvous_util.h
|
|
rendezvous_util_test.cc
|
|
replicate_per_replica_nodes.cc
|
|
replicate_per_replica_nodes.h
|
|
replicate_per_replica_nodes_test.cc
|
|
request_cost.cc
|
|
request_cost.h
|
|
request_cost_accessor.h
|
|
request_cost_accessor_registry.cc
|
|
request_cost_accessor_registry.h
|
|
request_cost_accessor_registry_test.cc
|
|
request_cost_test.cc
|
|
ring_alg.cc
|
|
ring_alg.h
|
|
ring_gatherer.cc
|
|
ring_gatherer.h
|
|
ring_gatherer_test.cc
|
|
ring_reducer.cc
|
|
ring_reducer.h
|
|
ring_reducer_test.cc
|
|
scoped_allocator.cc
|
|
scoped_allocator.h
|
|
scoped_allocator_mgr.cc
|
|
scoped_allocator_mgr.h
|
|
scoped_allocator_mgr_test.cc
|
|
session.cc
|
|
session_factory.cc
|
|
session_factory.h
|
|
session_options.cc
|
|
session_state.cc
|
|
session_test.cc
|
|
shape_refiner.cc
|
|
shape_refiner.h
|
|
shape_refiner_test.cc
|
|
shared_counter.h
|
|
simple_propagator_state.cc
|
|
simple_propagator_state.h
|
|
single_threaded_cpu_device.cc
|
|
single_threaded_cpu_device.h
|
|
single_threaded_executor.cc
|
|
single_threaded_executor.h
|
|
single_threaded_executor_test.cc
|
|
stats_publisher_interface.cc
|
|
stats_publisher_interface.h
|
|
step_stats_collector.cc
|
|
step_stats_collector.h
|
|
test_collective_executor_mgr.h
|
|
testlib_ops.cc
|
|
threadpool_device.cc
|
|
threadpool_device.h
|
|
threadpool_device_factory.cc
|
|
threadpool_device_test.cc
|
|
type_inference.cc
|
|
type_inference.h
|
|
type_inference_test.cc
|
|
zen_layout_pass.cc
|