Home
last modified time | relevance | path

Searched defs:rhs_op (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Ddeduplicate_batch_function.cc66 auto& rhs_op = std::get<1>(op_pair); in AreEquivalent() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dpred_test.cc38 XlaOp rhs_op = ConstantR0<bool>(&builder, rhs); in TestCompare() local
Dscalar_computations_test.cc52 XlaOp rhs_op = ConstantR0<NativeT>(&builder, rhs); in TestCompare() local
63 XlaOp rhs_op = ConstantR0<NativeT>(&builder, rhs); in TestMinMax() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
Dgemm_pattern.cc171 auto rhs_op = rhs_matrix.transpose ? CUBLAS_OP_T : CUBLAS_OP_N; in CreateTfrtOps() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc494 llvm::Value* rhs_op = in EmitTiledLlvmIrGemv() local