| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/ |
| H A 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()
779 CHECK(rhs_array.buffer) << "RHS must be constant"; in CompareArrayBuffers()
781 const auto& rhs_data = rhs_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()
805 const Array& rhs_array) { in HaveSameQuantizationParams() argument
806 if (lhs_array.quantization_params || rhs_array.quantization_params) { in HaveSameQuantizationParams()
[all …]
|
| H A D | tooling_util.h | 153 bool CompareConstantArrays(const Array& lhs_array, const Array& rhs_array);
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | dot_op_emitter.cc | 127 const llvm_ir::IrArray& rhs_array,
249 const llvm_ir::IrArray& rhs_array, const llvm_ir::IrArray* addend_array, in DotOpEmitter() argument
257 rhs_array_(rhs_array), in DotOpEmitter()
1239 const llvm_ir::IrArray& rhs_array, const llvm_ir::IrArray* addend_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
1342 CollapseFirstNDims(b, rhs_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()
1395 rhs_array, nullptr /*addend_array*/, in EmitBatchDotOperation()
[all …]
|
| H A D | dot_op_emitter.h | 63 const llvm_ir::IrArray& rhs_array,
|
| H A D | ir_emitter.cc | 885 llvm_ir::IrArray rhs_array(GetIrArrayFor(rhs)); in HandleDot() local
894 << llvm_ir::DumpToString(*rhs_array.GetBasePointer()); in HandleDot()
899 return EmitDotOperation(*dot, target_array, lhs_array, rhs_array, in HandleDot()
2214 llvm_ir::IrArray rhs_array( in HandleFusion() local
2220 *dot, target_array, lhs_array, rhs_array, &addend_array, in HandleFusion()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_evaluator_test.cc | 889 auto rhs_array = std::make_unique<Array2D<float>>(3, 2); in TEST_P() local
890 rhs_array->FillUnique(1.0f); in TEST_P()
891 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P()
933 auto rhs_array = std::make_unique<Array2D<float>>(3, 2); in TEST_P() local
934 rhs_array->FillUnique(1.0f); in TEST_P()
935 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P()
970 auto rhs_array = std::make_unique<Array4D<float>>(2, 2, 3, 1); in TEST_P() local
971 rhs_array->FillIota(2.0f); in TEST_P()
972 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*rhs_array); in TEST_P()
1014 Array3D<float> rhs_array = {{{3.f, 4.f}}}; in TEST_P() local
[all …]
|
| H A D | hlo_evaluator_typed_visitor.h | 1206 Array2D<NativeT> rhs_array(contracted_dimension_size,
1208 rhs_array.SetValues(rhs_literal.data<NativeT>());
1210 HloEvaluator::MatmulArray2D(lhs_array, rhs_array);
|
| /aosp_15_r20/external/tensorflow/tensorflow/security/advisory/ |
| H A D | tfsa-2021-100.md | 16 rhs_array = np.array([1,1])
17 rhs_tensor = tf.convert_to_tensor(np.reshape(rhs_array,(1,2)),dtype=tf.float32)
|
| H A D | tfsa-2021-039.md | 43 rhs_array = np.array([])
44 rhs_tensor = tf.convert_to_tensor(np.reshape(rhs_array,(0,1)),dtype=tf.float32)
|
| /aosp_15_r20/external/protobuf/src/google/protobuf/ |
| H A D | generated_message_reflection.cc | 538 uint32_t* rhs_array = r->MutableInlinedStringDonatedArray(rhs); in SwapInlinedStrings() local
540 uint32_t* rhs_state = &rhs_array[index / 32]; in SwapInlinedStrings()
542 bool rhs_arena_dtor_registered = (rhs_array[0] & 0x1u) == 0; in SwapInlinedStrings()
2537 uint32_t* rhs_array = MutableInlinedStringDonatedArray(rhs); in Swap() local
2539 GOOGLE_CHECK_EQ(rhs_array[0] & 0x1u, 0u); in Swap()
2545 ClearInlinedStringDonated(index, rhs_array); in Swap()
2548 SetInlinedStringDonated(index, rhs_array); in Swap()
|
| /aosp_15_r20/external/cronet/third_party/protobuf/src/google/protobuf/ |
| H A D | generated_message_reflection.cc | 538 uint32_t* rhs_array = r->MutableInlinedStringDonatedArray(rhs); in SwapInlinedStrings() local
540 uint32_t* rhs_state = &rhs_array[index / 32]; in SwapInlinedStrings()
542 bool rhs_arena_dtor_registered = (rhs_array[0] & 0x1u) == 0; in SwapInlinedStrings()
2537 uint32_t* rhs_array = MutableInlinedStringDonatedArray(rhs); in Swap() local
2539 GOOGLE_CHECK_EQ(rhs_array[0] & 0x1u, 0u); in Swap()
2545 ClearInlinedStringDonated(index, rhs_array); in Swap()
2548 SetInlinedStringDonated(index, rhs_array); in Swap()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | dot_operation_test.cc | 1185 Array2D<float> rhs_array({{5.0f, 6.0f}, {7.0f, 8.0f}}); in XLA_TEST_F() local
1186 auto rhs_constant = ConstantR2FromArray2D(&builder, rhs_array); in XLA_TEST_F()
|