Searched refs:get_output_index (Results 1 – 3 of 3) sorted by relevance
157 Optional<int32_t> get_output_index(AllocOp op) { in get_output_index() function in mlir::kernel_gen::transforms::__anon8364ced70111::BufferReuseAnalysis336 if (auto output_index = analysis.get_output_index(op)) { in runOnFunction()
489 TF_RETURN_IF_ERROR(get_output_index(output_name, &output_index)); in forward_input_to_output_with_shape()873 Status OpKernelContext::get_output_index(StringPiece name, in get_output_index() function in tensorflow::OpKernelContext889 TF_RETURN_IF_ERROR(get_output_index(name, &index)); in set_output()896 TF_RETURN_IF_ERROR(get_output_index(name, &index)); in set_output()1004 TF_RETURN_IF_ERROR(get_output_index(name, &index)); in set_output_ref()1011 TF_RETURN_IF_ERROR(get_output_index(name, &index)); in mutable_output()
1295 Status get_output_index(StringPiece name, int* out_index) const;