Searched refs:output_attr_array (Results 1 – 20 of 20) sorted by relevance
37 params.output_attr_array = runner.output_alloc_attrs().data(); in SetUpParams()
89 params.output_attr_array = output_attrs.data(); in EvaluateNode()
132 params.output_attr_array = output_attrs.data(); in EvaluateOperation()
78 params.output_attr_array = &alloc_attrs; in TestScalarSummaryOp()
66 params->output_attr_array = attrs->data(); in SetOutputAttrs()
315 op_params.output_attr_array = &generic_alloc_attr; in RunReduce()
153 params.output_attr_array = output_attr.data(); in Compile()
610 const AllocatorAttributes* output_attr_array = nullptr; member1043 return params_->output_attr_array[index]; in output_alloc_attr()
459 params.output_attr_array = output_alloc_attrs.data(); in TEST_F()
481 const auto output_attr = params_->output_attr_array == nullptr in forward_input_to_output_with_shape()
340 op_params.output_attr_array = &generic_alloc_attr; in RunCollective()
441 params.output_attr_array = kernel_state.output_alloc_attrs.data(); in Run()
856 params.output_attr_array = item.output_attrs(); in ProcessInline()
277 params.output_attr_array = output_alloc_attrs_.data(); in Run()
1296 p.output_attr_array = &alloc_attrs; in TEST_F()1368 p.output_attr_array = &alloc_attrs; in TEST_F()
450 params.output_attr_array = allocator_attributes.data(); in CallTfKernel()
421 run_state->params.output_attr_array = in ExecuteOpKernelRunner()
274 params->output_attr_array = attrs->data(); in SetOutputAttrs()
644 params_.output_attr_array = output_attr.data(); in LegalizeOp()
524 params->output_attr_array = allocator_attrs_.data(); in CreateOpKernelContext()