| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 746 bool CompareArrayBuffers(const Array& lhs_array, const Array& rhs_array) { in CompareArrayBuffers() argument
747 CHECK(lhs_array.data_type == rhs_array.data_type) << "Data types must match"; in CompareArrayBuffers()
748 CHECK(lhs_array.buffer) << "LHS must be constant"; in CompareArrayBuffers()
750 const auto& lhs_data = lhs_array.GetBuffer<A>().data; in CompareArrayBuffers()
762 bool HaveSameMinMax(const Array& lhs_array, const Array& rhs_array) { in HaveSameMinMax() argument
763 if (lhs_array.minmax || rhs_array.minmax) { in HaveSameMinMax()
764 if (!lhs_array.minmax || !rhs_array.minmax) { in HaveSameMinMax()
767 if (!(*lhs_array.minmax == *rhs_array.minmax)) { in HaveSameMinMax()
774 bool HaveSameQuantizationParams(const Array& lhs_array, in HaveSameQuantizationParams() argument
776 if (lhs_array.quantization_params || rhs_array.quantization_params) { in HaveSameQuantizationParams()
[all …]
|
| D | tooling_util.h | 155 bool CompareConstantArrays(const Array& lhs_array, const Array& rhs_array);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 111 const llvm_ir::IrArray& lhs_array,
207 const llvm_ir::IrArray& lhs_array, in DotOpEmitter() argument
217 lhs_array_(lhs_array), in DotOpEmitter()
847 const llvm_ir::IrArray& lhs_array, const llvm_ir::IrArray& rhs_array, in EmitNonBatchDotOperation() argument
856 target_array, lhs_array, rhs_array, addend_array, in EmitNonBatchDotOperation()
930 const llvm_ir::IrArray& lhs_array, const llvm_ir::IrArray& rhs_array, in EmitBatchDotOperation() argument
947 CollapseFirstNDims(b, lhs_array, num_batch_dims); in EmitBatchDotOperation()
1032 const llvm_ir::IrArray& lhs_array, in EmitDotOperation() argument
1045 return EmitBatchDotOperation(dot, target_array, lhs_array, rhs_array, in EmitDotOperation()
1051 lhs_array, rhs_array, addend_array, in EmitDotOperation()
|
| D | dot_op_emitter.h | 62 const llvm_ir::IrArray& lhs_array,
|
| D | ir_emitter.cc | 929 llvm_ir::IrArray lhs_array(GetIrArrayFor(lhs)); in HandleDot() local
937 << llvm_ir::DumpToString(*lhs_array.GetBasePointer()); in HandleDot()
944 return EmitDotOperation(*dot, target_array, lhs_array, rhs_array, in HandleDot()
2185 llvm_ir::IrArray lhs_array( in HandleFusion() local
2193 EmitDotOperation(*dot, target_array, lhs_array, rhs_array, in HandleFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | binop_scaling_test.cc | 115 Array4D<int> lhs_array({ in TEST_F() local
133 auto lhs = ConstantR4FromArray4D(&builder, lhs_array); in TEST_F()
|
| D | dot_operation_test.cc | 1141 Array2D<float> lhs_array({{1.0f, 2.0f}, {3.0f, 4.0f}}); in XLA_TEST_F() local
1142 auto lhs_constant = ConstantR2FromArray2D(&builder, lhs_array); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 471 const llvm_ir::IrArray& lhs_array = GetIrArray(*lhs_instruction, *dot); in HandleDot() local
486 lhs_array.EmitReadArrayElement(/*index=*/element_index, &b_); in HandleDot()
492 result = llvm::ConstantAggregateZero::get(lhs_array.GetElementLlvmType()); in HandleDot()
533 lhs_array, /*dimension_to_skip=*/lhs_reduction_dimension, "lhs"); in HandleDot()
567 llvm::Constant::getNullValue(lhs_array.GetElementLlvmType()), // init 0 in HandleDot()
579 llvm_ir::IrArray::Index lhs_index(lhs_multi_index, lhs_array.GetShape(), in HandleDot()
581 llvm::Value* lhs_element = lhs_array.EmitReadArrayElement(lhs_index, &b_); in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 722 auto lhs_array = absl::make_unique<Array2D<float>>(4, 1); in TEST_P() local
723 lhs_array->FillUnique(1.0f); in TEST_P()
724 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P()
806 auto lhs_array = absl::make_unique<Array2D<float>>(4, 3); in TEST_P() local
807 lhs_array->FillUnique(1.0f); in TEST_P()
808 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P()
849 auto lhs_array = absl::make_unique<Array4D<float>>(2, 2, 3, 1); in TEST_P() local
850 lhs_array->FillIota(1.0f); in TEST_P()
851 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*lhs_array); in TEST_P()
894 Array3D<float> lhs_array = {{{1, 2, 3}}}; in TEST_P() local
[all …]
|
| D | hlo_evaluator_typed_visitor.h | 1253 Array2D<NativeT> lhs_array(lhs->shape().dimensions(0), in HandleDot()
1255 lhs_array.SetValues(lhs_literal.data<NativeT>()); in HandleDot()
1260 HloEvaluator::MatmulArray2D(lhs_array, rhs_array); in HandleDot()
|