Searched refs:input_alloc_attrs (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_utils.cc | 36 params.input_alloc_attrs = runner.input_alloc_attrs(); in SetUpParams()
|
| D | op_kernel_runner.h | 95 const gtl::InlinedVector<AllocatorAttributes, 4>& input_alloc_attrs() const { in input_alloc_attrs() function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_executor.cc | 296 AllocatorAttributeVec input_alloc_attrs; in Run() local
418 input_alloc_attrs.clear(); in Run()
419 input_alloc_attrs.resize(num_inputs); in Run()
436 input_alloc_attrs[j] = input_alloc_attrs_[input_start_index + j]; in Run()
439 params.input_alloc_attrs = input_alloc_attrs; in Run()
|
| D | executor.cc | 315 AllocatorAttributeVec* input_alloc_attrs,
522 saved_input_alloc_attrs(p.input_alloc_attrs.begin(), in AsyncState()
523 p.input_alloc_attrs.end()), in AsyncState()
533 params.input_alloc_attrs = saved_input_alloc_attrs; in AsyncState()
721 AllocatorAttributeVec input_alloc_attrs; in ProcessInline() local
837 s = PrepareInputs(item, first_input, &inputs, &input_alloc_attrs, in ProcessInline()
860 params.input_alloc_attrs = input_alloc_attrs; in ProcessInline()
912 AllocatorAttributeVec* input_alloc_attrs, bool* is_input_dead) { in PrepareInputs() argument
914 input_alloc_attrs->resize(item.num_inputs); in PrepareInputs()
922 (*input_alloc_attrs)[i] = entry->alloc_attr; in PrepareInputs()
|
| D | collective_util.cc | 90 sub_params_.input_alloc_attrs = sub_input_attr_;
|
| D | collective_test_util.cc | 326 op_params.input_alloc_attrs = input_aa; in RunCollective()
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | op_kernel_runner.h | 76 const gtl::InlinedVector<AllocatorAttributes, 4>& input_alloc_attrs() const { in input_alloc_attrs() function
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 647 absl::Span<const AllocatorAttributes> input_alloc_attrs; member
1033 if (params_->input_alloc_attrs.empty()) { in input_alloc_attr()
1037 DCHECK_LT(index, params_->input_alloc_attrs.size()); in input_alloc_attr()
1038 return params_->input_alloc_attrs[index]; in input_alloc_attr()
|
| D | op_kernel.cc | 566 const auto input_attr = params_->input_alloc_attrs.empty() in forward_input()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 420 run_state->params.input_alloc_attrs = op_kernel_runner.input_alloc_attrs(); in ExecuteOpKernelRunner()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 276 params.input_alloc_attrs = input_alloc_attrs_; in Run()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 311 op_params.input_alloc_attrs = input_aa; in RunReduce()
|