Searched refs:GetOrCreateView (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | mhlo_to_lhlo_with_xla.cc | 245 TF_RETURN_IF_ERROR(GetOrCreateView(instr->operand(i), &operands, in CreateOperands()
250 GetOrCreateView(instr, &operands, /*result_subset=*/{}, token_mode)); in CreateOperands()
575 TF_RETURN_IF_ERROR(GetOrCreateView(instr, &output)); in EmitFusionOp()
866 TF_RETURN_IF_ERROR(GetOrCreateView(custom_call->operand(0), &operands)); in EmitCublasLtMatmul()
867 TF_RETURN_IF_ERROR(GetOrCreateView(custom_call->operand(1), &operands)); in EmitCublasLtMatmul()
868 TF_RETURN_IF_ERROR(GetOrCreateView( in EmitCublasLtMatmul()
870 TF_RETURN_IF_ERROR(GetOrCreateView(custom_call, &operands)); in EmitCublasLtMatmul()
873 TF_RETURN_IF_ERROR(GetOrCreateView( in EmitCublasLtMatmul()
1190 TF_RETURN_IF_ERROR(GetOrCreateView(operand, &operands)); in EmitAllReduceStartOp()
1192 TF_RETURN_IF_ERROR(GetOrCreateView(instr, &operands, /*result_subset=*/{})); in EmitAllReduceStartOp()
[all …]
|
| H A D | mhlo_to_lhlo_with_xla.h | 203 tensorflow::Status GetOrCreateView(
|