Searched refs:output_attr_array (Results 1 – 13 of 13) sorted by relevance
110 params.output_attr_array = output_attrs.data(); in EvaluateNode()
62 params->output_attr_array = gtl::vector_as_array(attrs); in SetOutputAttrs()
274 op_params.output_attr_array = &generic_alloc_attr; in DoReduce()
147 params.output_attr_array = output_attr.data(); in Compile()
259 params.output_attr_array = kernel_state.output_alloc_attrs.data(); in RunAsync()
206 params_->output_attr_array = gtl::vector_as_array(&allocator_attrs_); in CreateOpKernelContext()
238 params.output_attr_array = gtl::vector_as_array(&out_attrs); in Run()
620 const AllocatorAttributes* output_attr_array = nullptr; member1052 return params_->output_attr_array[index]; in output_alloc_attr()
452 const auto output_attr = params_->output_attr_array == nullptr in forward_input_to_output_with_shape()
465 op_params.output_attr_array = &generic_alloc_attr; in DoGather()
496 op_params.output_attr_array = &generic_alloc_attr; in DoReduce()
658 op_params.output_attr_array = &generic_alloc_attr; in DoBroadcast()
1734 params.output_attr_array = item.output_attrs(); in Process()