| /external/tensorflow/tensorflow/core/framework/ |
| memory_types.h | 33 MemoryTypeVector* input_memory_types,
|
| op_kernel.cc | 99 input_memory_types_(context->input_memory_types().begin(),
100 context->input_memory_types().end()),
204 const DataTypeSlice& input_types, const MemoryTypeSlice& input_memory_types,
215 input_memory_types_(input_memory_types),
1316 MemoryTypeVector input_memory_types; local
[all...] |
| op_kernel.h | 160 const MemoryTypeVector& input_memory_types() const { function in class:tensorflow::OpKernel
270 const MemoryTypeSlice& input_memory_types,
316 const MemoryTypeSlice& input_memory_types() const { function in class:tensorflow::OpKernelConstruction
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| create_xla_launch_op.cc | 178 MemoryTypeVector input_memory_types(fbody->arg_types.size(), DEVICE_MEMORY);
192 input_memory_types[i] = HOST_MEMORY;
234 &fbody->fdef.signature(), flr, fbody->arg_types, input_memory_types, local
|
| create_xla_launch_op_test.cc | 111 EXPECT_EQ(DEVICE_MEMORY, kernel_->input_memory_types()[0]);
112 EXPECT_EQ(HOST_MEMORY, kernel_->input_memory_types()[1]);
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| kernel_and_device.cc | 213 in_attrs[i].set_on_host(kernel_->input_memory_types()[i] ==
396 if (kernel_->input_memory_types()[i] == HOST_MEMORY) {
|
| /external/tensorflow/tensorflow/core/graph/ |
| graph_partition.cc | 550 MemoryTypeVector input_memory_types; local
564 &input_memory_types, &output_memory_types));
568 for (int i = 0; i < input_memory_types.size(); ++i) {
569 info->input_types[{node_id, i}] = input_memory_types[i];
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function.cc | 578 MemoryTypeVector input_memory_types; local
580 input_memory_types.push_back(MTypeFromDType(t));
591 &fbody->fdef.signature(), this, fbody->arg_types, input_memory_types,
[all...] |
