Searched defs:lhs_index (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
H A D | einsum.cc | 399 auto lhs_index = lhs_map.find(rhs[i]); in GetEinsumDimensionNumbers() local 410 auto lhs_index = lhs_map.find(out[i]); in GetEinsumDimensionNumbers() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
H A D | arithmetic.cc | 121 XlaOp lhs_index = in CreateMinMaxComputation() local
|
/aosp_15_r20/external/libtextclassifier/native/utils/grammar/utils/ |
H A D | rules.cc | 81 void LowerRule(const int lhs_index, const Rules::Rule& rule, in LowerRule()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
H A D | dot_op_emitter.cc | 665 llvm_ir::IrArray::Index lhs_index(lhs_multi_index, lhs_shape, in EmitNaiveLlvmIrGemm() local
|
/aosp_15_r20/art/runtime/ |
H A D | class_linker.cc | 7641 bool operator()(uint32_t lhs_index, ArtMethod* rhs) const NO_THREAD_SAFETY_ANALYSIS { in operator ()() 7646 bool operator()(uint32_t lhs_index, uint32_t rhs_index) const NO_THREAD_SAFETY_ANALYSIS { in operator ()() 7686 bool operator()(uint32_t lhs_index, ArtMethod* rhs) const NO_THREAD_SAFETY_ANALYSIS { in operator ()() 7693 bool operator()(uint32_t lhs_index, uint32_t rhs_index) const NO_THREAD_SAFETY_ANALYSIS { in operator ()()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | elemental_ir_emitter.cc | 2410 IrArray::Index lhs_index(lhs_multi_index, hlo->operand(0)->shape(), in EmitElementalDot() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
H A D | tfl_ops.cc | 572 const int64_t lhs_index = in ConstFoldBinaryOpDenseDense() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
H A D | legalize_tf.cc | 821 Value lhs_index = block->getArgument(1); in BuildArgMinMaxReductionBody() local
|