Searched refs:result_op (Results 1 – 7 of 7) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_intdiv.cpp | 103 int32_t result_op = divider.divide(num); in test_powers_32bit() local
104 VERIFY_IS_EQUAL(result_op, result); in test_powers_32bit()
121 int64_t result_op = divider.divide(num); in test_powers_64bit() local
122 VERIFY_IS_EQUAL(result_op, result); in test_powers_64bit()
134 int64_t result_op = divider.divide(num); in test_specific() local
135 VERIFY_IS_EQUAL(result, result_op); in test_specific()
|
| /external/tensorflow/tensorflow/python/autograph/converters/ |
| D | functions_test.py | 39 result_op = tr(constant_op.constant(1))
40 self.assertIn('f/', result_op.op.name)
57 result_op = tr()
58 self.assertIn('f/', result_op.op.name)
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_all_reduce_thunk.cc | 103 mlir::Operation* result_op = result.getDefiningOp(); in FindReductionOp() local
104 TF_RET_CHECK(result_op); in FindReductionOp()
107 if (mlir::isa<mlir::mhlo::FusionOp>(result_op)) { in FindReductionOp()
108 return FindReductionOp(result_op->getRegion(0).front()); in FindReductionOp()
112 if (absl::c_is_permutation(result_op->getOperands(), block.getArguments())) { in FindReductionOp()
113 return result_op; in FindReductionOp()
117 TF_RET_CHECK(mlir::isa<mlir::mhlo::ConvertOp>(result_op)); in FindReductionOp()
118 TF_RET_CHECK(result_op->getNumOperands() == 1); in FindReductionOp()
119 mlir::Operation* reduction_op = result_op->getOperand(0).getDefiningOp(); in FindReductionOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/ |
| D | lift_as_function_call_utils.cc | 183 Operation *result_op = results[0].getDefiningOp(); in LiftAsFunctionCall() local
184 auto module = result_op->getParentOfType<ModuleOp>(); in LiftAsFunctionCall()
187 auto current_func = result_op->getParentOfType<func::FuncOp>(); in LiftAsFunctionCall()
226 builder.setInsertionPointAfter(result_op); in LiftAsFunctionCall()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | merge_control_flow.cc | 241 Operation* result_op, Operation* after_op, in GetMoveOpsBetweenTwoIfRegions() argument
249 func::FuncOp func = result_op->getParentOfType<func::FuncOp>(); in GetMoveOpsBetweenTwoIfRegions()
272 enqueue_deps(result_op); in GetMoveOpsBetweenTwoIfRegions()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 491 llvm::Value* result_op = target_array_.GetBasePointer(); in EmitTiledLlvmIrGemv() local
518 /*result=*/result_op, b_, hlo_module_config_); in EmitTiledLlvmIrGemv()
528 /*result=*/result_op, b_, hlo_module_config_); in EmitTiledLlvmIrGemv()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2703 Operation *result_op = AvgPoolDivideByCount<OpTy, num_dims>( in matchAndRewrite() local
2707 Value result = result_op->getOpResult(0); in matchAndRewrite()
|