Searched refs:allocator_used (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 415 se::ScratchAllocator* allocator_used = in operator ()() local
422 input_desc, &in_backprop_ptr_rz, allocator_used, in operator ()()
|
| D | conv_ops_3d.cc | 535 se::ScratchAllocator* allocator_used = in launch() local
542 output_desc, &output_ptr_rz, allocator_used, in launch()
|
| D | conv_ops_fused_impl.h | 399 se::ScratchAllocator* allocator_used =
406 launch(se::dnn::AlgorithmConfig(profile_algorithm), allocator_used,
|
| D | conv_grad_filter_ops.cc | 1009 se::ScratchAllocator* allocator_used = in operator ()() local
1017 filter_desc, &filter_backprop_ptr_rz, allocator_used, in operator ()()
|
| D | conv_ops.cc | 1021 se::ScratchAllocator* allocator_used = in operator ()() local
1029 output_desc, &output_tensor, allocator_used, in operator ()()
|
| D | conv_grad_ops_3d.cc | 1449 se::ScratchAllocator* allocator_used = in Compute() local
1456 input_desc, &in_backprop_ptr_rz, allocator_used, in Compute()
|