| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types_test.cc | 71 MemoryTypeVector({DEVICE_MEMORY, DEVICE_MEMORY, HOST_MEMORY, HOST_MEMORY, in TEST()
72 HOST_MEMORY, DEVICE_MEMORY, DEVICE_MEMORY, in TEST()
73 DEVICE_MEMORY, HOST_MEMORY, HOST_MEMORY, HOST_MEMORY}), in TEST()
83 MemoryTypeVector({HOST_MEMORY, DEVICE_MEMORY, HOST_MEMORY, HOST_MEMORY, in TEST()
84 HOST_MEMORY, HOST_MEMORY, HOST_MEMORY, HOST_MEMORY, in TEST()
85 HOST_MEMORY, HOST_MEMORY, HOST_MEMORY}), in TEST()
87 EXPECT_EQ(MemoryTypeVector({HOST_MEMORY, HOST_MEMORY, HOST_MEMORY, in TEST()
|
| D | memory_types.cc | 52 (*memory_types)[j] = HOST_MEMORY; in MemoryTypesHelper()
71 return (dtype == DT_INT32 || DataTypeAlwaysOnHost(dtype)) ? HOST_MEMORY in MTypeFromDType()
133 (*inp_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
138 (*out_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
146 (*inp_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
153 (*out_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
|
| D | types.h | 49 HOST_MEMORY = 1, enumerator
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | memory_types.cc | 163 if (((sm == HOST_MEMORY) && (dm == DEVICE_MEMORY)) || in EnsureMemoryTypes()
164 ((sm == DEVICE_MEMORY) && (dm == HOST_MEMORY))) { in EnsureMemoryTypes()
188 Send(g, tensor_name, device_name, (item.sm == HOST_MEMORY), e); in EnsureMemoryTypes()
189 recv = Recv(g, tensor_name, device_name, (item.dm == HOST_MEMORY), e); in EnsureMemoryTypes()
|
| D | memory_types_test.cc | 87 EXPECT_EQ(memory_type, HOST_MEMORY); in TEST()
93 EXPECT_EQ(memory_type, HOST_MEMORY); in TEST()
|
| D | partitioning_utils.cc | 104 if (MTypeFromDType(type) == HOST_MEMORY) { in UpdateArgAndRetvalMetadata()
115 if (MTypeFromDType(type) == HOST_MEMORY) { in UpdateArgAndRetvalMetadata()
|
| D | constant_folding.cc | 504 if ((memory_type == HOST_MEMORY && !is_int32) || in ReplaceTensorWithConstant()
|
| D | function.cc | 982 if (MTypeFromDType(arg_type) == HOST_MEMORY) { in RunRemote()
989 if (MTypeFromDType(ret_type) == HOST_MEMORY) { in RunRemote()
|
| D | executor.cc | 770 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in SetAllocAttrs()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 214 tensorflow::HOST_MEMORY); in Run()
219 tensorflow::HOST_MEMORY); in Run()
360 if (kernel_->output_memory_types()[idx] == HOST_MEMORY) { in OutputDevice()
396 if (kernel_->input_memory_types()[i] == HOST_MEMORY) { in InputDevice()
|
| D | execute.cc | 340 } else if (MTypeFromDType(tensor_handle->dtype) == HOST_MEMORY) { in AddInputDevicesToCacheKey()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op.cc | 192 input_memory_types[i] = HOST_MEMORY; in CreateXlaLaunchOp()
220 output_memory_types[i] = HOST_MEMORY; in CreateXlaLaunchOp()
|
| D | create_xla_launch_op_test.cc | 112 EXPECT_EQ(HOST_MEMORY, kernel_->input_memory_types()[1]); in TEST_F()
|
| D | partially_decluster_pass.cc | 86 dst_input_mtypes[e->dst_input()] == HOST_MEMORY; in FindNodesToDecluster()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 143 if (MTypeFromDType(dtype) == HOST_MEMORY) { in FillOutputDevices()
195 } else if (MTypeFromDType(dtype) == HOST_MEMORY) { in Instantiate()
|
| D | ops_testutil.h | 58 (params->op_kernel->output_memory_types()[index] == HOST_MEMORY); in SetOutputAttrs()
|
| D | pad_op.cc | 118 alloc_attrs.set_on_host(context->input_memory_type(0) == HOST_MEMORY); in Compute()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 153 return dst_it->second == HOST_MEMORY; in IsDstInputOnHost()
206 host_memory = (src_it->second == HOST_MEMORY); in AddSend()
269 host_memory = (dst_it->second == HOST_MEMORY); in AddRecv()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor.cc | 129 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in Initialize()
|
| D | dataset_test_base.cc | 202 (params_->op_kernel->output_memory_types()[index] == HOST_MEMORY); in CreateOpKernelContext()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_graph_utils.cc | 151 graph, device_type, memory_type == HOST_MEMORY, src_node->name(), in InsertNodes()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster.i | 262 if (inp_mtypes[i] == tensorflow::HOST_MEMORY) { in TF_GetSupportedDevices()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 384 if (out_mtypes[output_port] == HOST_MEMORY) { in IsHostMemory()
|