Searched refs:output_attr_array (Results 1 – 19 of 19) sorted by relevance
89 params.output_attr_array = output_attrs.data(); in EvaluateNode()
78 params.output_attr_array = &alloc_attrs; in TestScalarSummaryOp()
66 params->output_attr_array = attrs->data(); in SetOutputAttrs()
313 op_params.output_attr_array = &generic_alloc_attr; in RunReduce()
153 params.output_attr_array = output_attr.data(); in Compile()
255 params.output_attr_array = output_alloc_attrs_.data(); in Run()
410 params.output_attr_array = kernel_state.output_alloc_attrs.data(); in Run()
519 params->output_attr_array = allocator_attrs_.data(); in CreateOpKernelContext()
393 op_params.output_attr_array = &generic_alloc_attr; in DoPermute()
471 op_params.output_attr_array = &generic_alloc_attr; in DoGather()
505 op_params.output_attr_array = &generic_alloc_attr; in DoReduce()
656 op_params.output_attr_array = &generic_alloc_attr; in DoBroadcast()
803 params.output_attr_array = item.output_attrs(); in Process()
637 const AllocatorAttributes* output_attr_array = nullptr; member1087 return params_->output_attr_array[index]; in output_alloc_attr()
456 params.output_attr_array = output_alloc_attrs.data(); in TEST_F()
468 const auto output_attr = params_->output_attr_array == nullptr in forward_input_to_output_with_shape()
274 params->output_attr_array = attrs->data(); in SetOutputAttrs()
504 params_.output_attr_array = output_attr.data(); in LegalizeOp()
960 p.output_attr_array = &alloc_attrs; in TEST_F()