| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types_test.cc | 77 MemoryTypeVector({HOST_MEMORY, DEVICE_MEMORY, DEVICE_MEMORY, HOST_MEMORY, in TEST()
78 HOST_MEMORY, HOST_MEMORY, DEVICE_MEMORY, DEVICE_MEMORY, in TEST()
79 DEVICE_MEMORY, HOST_MEMORY, HOST_MEMORY, HOST_MEMORY}), in TEST()
84 HOST_MEMORY, HOST_MEMORY, DEVICE_MEMORY}), in TEST()
90 MemoryTypeVector({HOST_MEMORY, HOST_MEMORY, DEVICE_MEMORY, HOST_MEMORY, in TEST()
91 HOST_MEMORY, HOST_MEMORY, HOST_MEMORY, HOST_MEMORY, in TEST()
92 HOST_MEMORY, HOST_MEMORY, HOST_MEMORY, HOST_MEMORY}), in TEST()
95 HOST_MEMORY, HOST_MEMORY, HOST_MEMORY, in TEST()
96 HOST_MEMORY, HOST_MEMORY, DEVICE_MEMORY}), in TEST()
|
| D | memory_types.cc | 54 (*memory_types)[j] = HOST_MEMORY; in MemoryTypesHelper()
74 return (dtype == DT_INT32 || DataTypeAlwaysOnHost(dtype)) ? HOST_MEMORY in MTypeFromDType()
79 return DataTypeAlwaysOnHost(dtype) ? HOST_MEMORY : DEVICE_MEMORY; in MTypeFromDTypeIntsOnDevice()
148 (*inp_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
153 (*out_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
161 (*inp_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
169 (*out_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
|
| D | types.h | 50 HOST_MEMORY = 1, enumerator
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | memory_types.cc | 169 if (((sm == HOST_MEMORY) && (dm == DEVICE_MEMORY)) || in EnsureMemoryTypes()
170 ((sm == DEVICE_MEMORY) && (dm == HOST_MEMORY))) { in EnsureMemoryTypes()
194 Send(g, tensor_name, device_name, (item.sm == HOST_MEMORY), e); in EnsureMemoryTypes()
195 recv = Recv(g, tensor_name, device_name, (item.dm == HOST_MEMORY), e); in EnsureMemoryTypes()
|
| D | partitioning_utils.cc | 210 if (mtype == HOST_MEMORY) { in UpdateArgAndRetvalMetadata()
225 if (mtype == HOST_MEMORY) { in UpdateArgAndRetvalMetadata()
|
| D | memory_types_test.cc | 75 EXPECT_EQ(memory_type, HOST_MEMORY); in TEST()
|
| D | node_file_writer.cc | 151 context->input_memory_type(i) == HOST_MEMORY) { in RecordNodeExecution()
|
| D | single_threaded_executor.cc | 203 kernel_state.kernel->output_memory_types()[0] == HOST_MEMORY; in Initialize()
229 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in Initialize()
|
| D | graph_view.cc | 449 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in SetAllocAttrs()
|
| D | constant_folding.cc | 528 if ((memory_type == HOST_MEMORY && !is_int32) || in ReplaceTensorWithConstant()
|
| D | function.cc | 1065 if (MTypeFromDType(arg_type) == HOST_MEMORY) { in RunRemote()
1072 if (MTypeFromDType(ret_type) == HOST_MEMORY) { in RunRemote()
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | op_kernel_runner.cc | 148 tensorflow::HOST_MEMORY); in OpKernelRunner()
154 tensorflow::HOST_MEMORY); in OpKernelRunner()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | op_kernel_runner.cc | 145 tensorflow::HOST_MEMORY); in OpKernelRunner()
151 tensorflow::HOST_MEMORY); in OpKernelRunner()
|
| D | kernel_fallback_op_handler.cc | 140 if (op_kernel->output_memory_types()[index] == MemoryType::HOST_MEMORY && in GetResultDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 124 bool host = kernel_->input_memory_types()[i] == tensorflow::HOST_MEMORY; in Init()
133 tensorflow::HOST_MEMORY); in Init()
482 if (kernel_->output_memory_types()[idx] == HOST_MEMORY) { in OutputDevice()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator_test.cc | 120 EXPECT_EQ(HOST_MEMORY, kernel_->input_memory_types()[1]); in TEST_F()
|
| D | xla_compile_on_demand_op.cc | 119 return ctx->input_memory_type(idx) == HOST_MEMORY; in Compile()
|
| D | compilability_check_util.cc | 686 input_memory_types[i] = tensorflow::HOST_MEMORY; in GetInputMemoryTypes()
698 output_memory_types[i] = tensorflow::HOST_MEMORY; in GetOutputMemoryTypes()
|
| D | partially_decluster_pass.cc | 98 dst_input_mtypes[e->dst_input()] == HOST_MEMORY; in FindNodesToDecluster()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 155 return dst_it->second == HOST_MEMORY; in IsDstInputOnHost()
209 host_memory = (src_it->second == HOST_MEMORY); in AddSend()
273 host_memory = (dst_it->second == HOST_MEMORY); in AddRecv()
279 src_host_memory = (src_it->second == HOST_MEMORY); in AddRecv()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 62 (params->op_kernel->output_memory_types()[index] == HOST_MEMORY); in SetOutputAttrs()
|
| D | pad_op.cc | 111 alloc_attrs.set_on_host(context->input_memory_type(0) == HOST_MEMORY); in Compute()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels.cc | 608 return cc_ctx->input_memory_type(i) == tensorflow::HOST_MEMORY; in TF_IsHostMemoryInput()
618 return cc_ctx->output_memory_type(i) == tensorflow::HOST_MEMORY; in TF_IsHostMemoryOutput()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster_wrapper.cc | 202 if (inp_mtypes[i] == tensorflow::HOST_MEMORY) { in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_graph_utils.cc | 184 CreateCopyNode(graph, device_type, memory_type == HOST_MEMORY, in InsertNodes()
|