|
..
|
|
registration
|
|
BUILD
|
|
allocation_description.proto
|
|
allocator.cc
|
|
allocator.h
|
|
allocator_registry.cc
|
|
allocator_registry.h
|
|
allocator_test.cc
|
|
api_def.proto
|
|
attr_value.proto
|
|
attr_value_util.cc
|
|
attr_value_util.h
|
|
attr_value_util_test.cc
|
|
batch_util_test.cc
|
|
bfloat16.cc
|
|
bfloat16.h
|
|
bfloat16_test.cc
|
|
bounds_check.h
|
|
cancellation.cc
|
|
cancellation.h
|
|
cancellation_test.cc
|
|
collective.cc
|
|
collective.h
|
|
common_shape_fns.cc
|
|
common_shape_fns.h
|
|
common_shape_fns_test.cc
|
|
control_flow.h
|
|
cost_graph.proto
|
|
cpu_allocator_impl.cc
|
|
dataset.cc
|
|
dataset.h
|
|
dataset_metadata.proto
|
|
dataset_options.proto
|
|
dataset_stateful_op_allowlist.h
|
|
dataset_test.cc
|
|
device.cc
|
|
device.h
|
|
device_attributes.proto
|
|
device_base.cc
|
|
device_base.h
|
|
device_base_test.cc
|
|
device_factory.cc
|
|
device_factory.h
|
|
disable_jit_test.cc
|
|
fake_input.cc
|
|
fake_input.h
|
|
full_type.proto
|
|
full_type_inference_util.cc
|
|
full_type_inference_util.h
|
|
full_type_inference_util_test.cc
|
|
full_type_util.cc
|
|
full_type_util.h
|
|
full_type_util_test.cc
|
|
function.cc
|
|
function.h
|
|
function.proto
|
|
function_handle_cache.cc
|
|
function_handle_cache.h
|
|
function_test.cc
|
|
function_testlib.cc
|
|
function_testlib.h
|
|
graph.proto
|
|
graph_def_util.cc
|
|
graph_def_util.h
|
|
graph_def_util_test.cc
|
|
graph_to_functiondef.cc
|
|
graph_to_functiondef.h
|
|
graph_to_functiondef_test.cc
|
|
graph_transfer_info.proto
|
|
kernel_def.proto
|
|
kernel_def_builder.cc
|
|
kernel_def_builder.h
|
|
kernel_def_builder_test.cc
|
|
kernel_def_util.cc
|
|
kernel_def_util.h
|
|
kernel_def_util_test.cc
|
|
kernel_shape_util.cc
|
|
kernel_shape_util.h
|
|
load_library.cc
|
|
local_rendezvous.cc
|
|
local_rendezvous.h
|
|
log_memory.cc
|
|
log_memory.h
|
|
log_memory.proto
|
|
logging.cc
|
|
logging.h
|
|
lookup_interface.cc
|
|
lookup_interface.h
|
|
memory_types.cc
|
|
memory_types.h
|
|
memory_types_test.cc
|
|
metrics.cc
|
|
metrics.h
|
|
model.cc
|
|
model.h
|
|
model.proto
|
|
model_test.cc
|
|
node_def.proto
|
|
node_def_builder.cc
|
|
node_def_builder.h
|
|
node_def_builder_test.cc
|
|
node_def_util.cc
|
|
node_def_util.h
|
|
node_def_util_test.cc
|
|
node_properties.cc
|
|
node_properties.h
|
|
node_properties_test.cc
|
|
numeric_op.h
|
|
numeric_types.h
|
|
op.cc
|
|
op.h
|
|
op_compatibility_test.cc
|
|
op_def.proto
|
|
op_def_builder.cc
|
|
op_def_builder.h
|
|
op_def_builder_test.cc
|
|
op_def_util.cc
|
|
op_def_util.h
|
|
op_def_util_test.cc
|
|
op_gen_lib.cc
|
|
op_gen_lib.h
|
|
op_gen_lib_test.cc
|
|
op_kernel.cc
|
|
op_kernel.h
|
|
op_kernel_test.cc
|
|
op_kernel_test_base.h
|
|
op_registration_test.cc
|
|
op_requires.h
|
|
op_segment.cc
|
|
op_segment.h
|
|
op_segment_test.cc
|
|
ops_util.cc
|
|
ops_util.h
|
|
partial_tensor_shape.h
|
|
partial_tensor_shape_test.cc
|
|
queue_interface.h
|
|
reader_base.cc
|
|
reader_base.h
|
|
reader_base.proto
|
|
reader_interface.h
|
|
reader_op_kernel.h
|
|
ref_var.cc
|
|
ref_var.h
|
|
register_types.h
|
|
register_types_traits.h
|
|
rendezvous.cc
|
|
rendezvous.h
|
|
rendezvous_test.cc
|
|
resource_base.h
|
|
resource_handle.cc
|
|
resource_handle.h
|
|
resource_handle.proto
|
|
resource_handle_test.cc
|
|
resource_mgr.cc
|
|
resource_mgr.h
|
|
resource_mgr_test.cc
|
|
resource_op_kernel.h
|
|
resource_op_kernel_test.cc
|
|
resource_var.cc
|
|
resource_var.h
|
|
resource_var_test.cc
|
|
rng_alg.h
|
|
run_handler.cc
|
|
run_handler.h
|
|
run_handler_test.cc
|
|
run_handler_util.cc
|
|
run_handler_util.h
|
|
run_handler_util_test.cc
|
|
session_state.h
|
|
shape_inference.cc
|
|
shape_inference.h
|
|
shape_inference_test.cc
|
|
shape_inference_testutil.cc
|
|
shape_inference_testutil.h
|
|
shape_inference_testutil_test.cc
|
|
shared_ptr_variant.h
|
|
stats_aggregator.h
|
|
step_stats.proto
|
|
summary.proto
|
|
tensor.cc
|
|
tensor.h
|
|
tensor.proto
|
|
tensor_description.proto
|
|
tensor_key.h
|
|
tensor_reference.h
|
|
tensor_shape.cc
|
|
tensor_shape.h
|
|
tensor_shape.proto
|
|
tensor_shape_test.cc
|
|
tensor_slice.cc
|
|
tensor_slice.h
|
|
tensor_slice.proto
|
|
tensor_slice_test.cc
|
|
tensor_test.cc
|
|
tensor_testutil.cc
|
|
tensor_testutil.h
|
|
tensor_testutil_test.cc
|
|
tensor_types.h
|
|
tensor_util.cc
|
|
tensor_util.h
|
|
tensor_util_test.cc
|
|
thread_factory.h
|
|
tracking_allocator.cc
|
|
tracking_allocator.h
|
|
tracking_allocator_test.cc
|
|
type_index.h
|
|
type_traits.h
|
|
typed_allocator.cc
|
|
typed_allocator.h
|
|
types.cc
|
|
types.h
|
|
types.proto
|
|
types_test.cc
|
|
variable.proto
|
|
variant.cc
|
|
variant.h
|
|
variant_encode_decode.h
|
|
variant_op_copy_test.cc
|
|
variant_op_registry.cc
|
|
variant_op_registry.h
|
|
variant_op_registry_test.cc
|
|
variant_tensor_data.cc
|
|
variant_tensor_data.h
|
|
variant_test.cc
|
|
versions.cc
|
|
versions.h
|
|
versions.proto
|