| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 776 bool CompareArrayBuffers(const Array& lhs_array, const Array& rhs_array) { in CompareArrayBuffers() argument
777 CHECK(lhs_array.data_type == rhs_array.data_type) << "Data types must match"; in CompareArrayBuffers()
778 CHECK(lhs_array.buffer) << "LHS must be constant"; in CompareArrayBuffers()
780 const auto& lhs_data = lhs_array.GetBuffer<A>().data; in CompareArrayBuffers()
792 bool HaveSameMinMax(const Array& lhs_array, const Array& rhs_array) { in HaveSameMinMax() argument
793 if (lhs_array.minmax || rhs_array.minmax) { in HaveSameMinMax()
794 if (!lhs_array.minmax || !rhs_array.minmax) { in HaveSameMinMax()
797 if (!(*lhs_array.minmax == *rhs_array.minmax)) { in HaveSameMinMax()
804 bool HaveSameQuantizationParams(const Array& lhs_array, in HaveSameQuantizationParams() argument
806 if (lhs_array.quantization_params || rhs_array.quantization_params) { in HaveSameQuantizationParams()
[all …]
|
| D | tooling_util.h | 153 bool CompareConstantArrays(const Array& lhs_array, const Array& rhs_array);
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_tensorflow_matmul.cc | 84 Array& lhs_array = model->GetArray(input_lhs); in Run() local
85 if (!lhs_array.has_shape()) { in Run()
93 int dimensions_count = lhs_array.shape().dimensions_count(); in Run()
|
| /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 | 1182 Array2D<float> lhs_array({{1.0f, 2.0f}, {3.0f, 4.0f}}); in XLA_TEST_F() local
1183 auto lhs_constant = ConstantR2FromArray2D(&builder, lhs_array); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 126 const llvm_ir::IrArray& lhs_array,
248 const llvm_ir::IrArray& target_array, const llvm_ir::IrArray& lhs_array, in DotOpEmitter() argument
256 lhs_array_(lhs_array), in DotOpEmitter()
1238 const llvm_ir::IrArray& target_array, const llvm_ir::IrArray& lhs_array, in EmitNonBatchDotOperation() argument
1249 target_array, lhs_array, rhs_array, addend_array, in EmitNonBatchDotOperation()
1324 const llvm_ir::IrArray& lhs_array, const llvm_ir::IrArray& rhs_array, in PotentiallyImplementedAsEigenMatmul() argument
1340 CollapseFirstNDims(b, lhs_array, num_batch_dims); in PotentiallyImplementedAsEigenMatmul()
1380 const llvm_ir::IrArray& lhs_array, const llvm_ir::IrArray& rhs_array, in EmitBatchDotOperation() argument
1391 dot, target_array, lhs_array, rhs_array, executable_run_options_value, in EmitBatchDotOperation()
1394 DotOpEmitter dot_emitter(dot_info, dot.name(), target_array, lhs_array, in EmitBatchDotOperation()
[all …]
|
| D | dot_op_emitter.h | 62 const llvm_ir::IrArray& lhs_array,
|
| D | ir_emitter.cc | 884 llvm_ir::IrArray lhs_array(GetIrArrayFor(lhs)); in HandleDot() local
892 << llvm_ir::DumpToString(*lhs_array.GetBasePointer()); in HandleDot()
899 return EmitDotOperation(*dot, target_array, lhs_array, rhs_array, in HandleDot()
2212 llvm_ir::IrArray lhs_array( in HandleFusion() local
2220 *dot, target_array, lhs_array, rhs_array, &addend_array, in HandleFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 837 auto lhs_array = std::make_unique<Array2D<float>>(4, 1); in TEST_P() local
838 lhs_array->FillUnique(1.0f); in TEST_P()
839 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P()
921 auto lhs_array = std::make_unique<Array2D<float>>(4, 3); in TEST_P() local
922 lhs_array->FillUnique(1.0f); in TEST_P()
923 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P()
964 auto lhs_array = std::make_unique<Array4D<float>>(2, 2, 3, 1); in TEST_P() local
965 lhs_array->FillIota(1.0f); in TEST_P()
966 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*lhs_array); in TEST_P()
1009 Array3D<float> lhs_array = {{{1, 2, 3}}}; in TEST_P() local
[all …]
|
| D | hlo_evaluator_typed_visitor.h | 1203 Array2D<NativeT> lhs_array(lhs->shape().dimensions(0),
1205 lhs_array.SetValues(lhs_literal.data<NativeT>());
1210 HloEvaluator::MatmulArray2D(lhs_array, rhs_array);
|
| /external/cronet/stable/third_party/protobuf/src/google/protobuf/ |
| D | generated_message_reflection.cc | 537 uint32_t* lhs_array = r->MutableInlinedStringDonatedArray(lhs); in SwapInlinedStrings() local
539 uint32_t* lhs_state = &lhs_array[index / 32]; in SwapInlinedStrings()
541 bool lhs_arena_dtor_registered = (lhs_array[0] & 0x1u) == 0; in SwapInlinedStrings()
2536 uint32_t* lhs_array = MutableInlinedStringDonatedArray(lhs); in Swap() local
2538 GOOGLE_CHECK_EQ(lhs_array[0] & 0x1u, 0u); in Swap()
2544 SetInlinedStringDonated(index, lhs_array); in Swap()
2547 ClearInlinedStringDonated(index, lhs_array); in Swap()
|
| /external/cronet/tot/third_party/protobuf/src/google/protobuf/ |
| D | generated_message_reflection.cc | 537 uint32_t* lhs_array = r->MutableInlinedStringDonatedArray(lhs); in SwapInlinedStrings() local
539 uint32_t* lhs_state = &lhs_array[index / 32]; in SwapInlinedStrings()
541 bool lhs_arena_dtor_registered = (lhs_array[0] & 0x1u) == 0; in SwapInlinedStrings()
2536 uint32_t* lhs_array = MutableInlinedStringDonatedArray(lhs); in Swap() local
2538 GOOGLE_CHECK_EQ(lhs_array[0] & 0x1u, 0u); in Swap()
2544 SetInlinedStringDonated(index, lhs_array); in Swap()
2547 ClearInlinedStringDonated(index, lhs_array); in Swap()
|
| /external/protobuf/src/google/protobuf/ |
| D | generated_message_reflection.cc | 537 uint32_t* lhs_array = r->MutableInlinedStringDonatedArray(lhs); in SwapInlinedStrings() local
539 uint32_t* lhs_state = &lhs_array[index / 32]; in SwapInlinedStrings()
541 bool lhs_arena_dtor_registered = (lhs_array[0] & 0x1u) == 0; in SwapInlinedStrings()
2536 uint32_t* lhs_array = MutableInlinedStringDonatedArray(lhs); in Swap() local
2538 GOOGLE_CHECK_EQ(lhs_array[0] & 0x1u, 0u); in Swap()
2544 SetInlinedStringDonated(index, lhs_array); in Swap()
2547 ClearInlinedStringDonated(index, lhs_array); in Swap()
|