| /external/tensorflow/tensorflow/core/kernels/ |
| constant_op_test.cc | 60 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, device.get(),
|
| ops_testutil.h | 104 kernel_ = CreateOpKernel(device_type_, device_.get(), allocator(),
|
| parse_tensor_test.cc | 50 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(),
|
| segment_reduction_ops_test.cc | 69 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(),
|
| sparse_to_dense_op_test.cc | 246 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(),
|
| restore_op_test.cc | 88 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(),
392 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(),
|
| restore_v2_op_test.cc | 100 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(), save,
|
| nn_ops_test.cc | [all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| op_kernel_test.cc | 139 std::unique_ptr<OpKernel> op(CreateOpKernel(
156 CreateOpKernel(std::move(device_type), &device_, cpu_allocator(),
295 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(),
320 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(),
348 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(),
403 // Test CreateOpKernel()
404 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, &device,
437 // Test CreateOpKernel().
438 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, &device,
629 std::unique_ptr<OpKernel> op(CreateOpKernel(
[all...] |
| op_segment_test.cc | 67 auto created = CreateOpKernel(DEVICE_CPU, &device_, cpu_allocator(),
|
| resource_op_kernel_test.cc | 99 std::unique_ptr<OpKernel> op(CreateOpKernel(
|
| op_kernel.cc | [all...] |
| op_kernel.h | [all...] |
| /external/tensorflow/tensorflow/c/eager/ |
| runtime.cc | 243 Status s = CreateOpKernel(device->device_type().c_str(), device,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| constant_folding.cc | 866 CreateOpKernel("CPU", cpu_device_, cpu_device_->GetAllocator({}), node,
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| executor.cc | [all...] |
