Home
last modified time | relevance | path

Searched refs:GetShape (Results 1 – 25 of 219) sorted by relevance

123456789

/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression_test.cc92 BuildInterpreter({GetShape(input_boxes_), GetShape(input_scores_), in NonMaxSuppressionV4OpModel()
93 GetShape(input_max_output_size_), in NonMaxSuppressionV4OpModel()
94 GetShape(input_iou_threshold_), in NonMaxSuppressionV4OpModel()
95 GetShape(input_score_threshold_)}); in NonMaxSuppressionV4OpModel()
186 {GetShape(input_boxes_), GetShape(input_scores_), in NonMaxSuppressionV5OpModel()
187 GetShape(input_max_output_size_), GetShape(input_iou_threshold_), in NonMaxSuppressionV5OpModel()
188 GetShape(input_score_threshold_), GetShape(input_sigma_)}); in NonMaxSuppressionV5OpModel()
Dquant_basic_lstm_test.cc114 BuildInterpreter({GetShape(input_), GetShape(zero_input_), in QuantizedLSTMOpModel()
115 GetShape(prev_output_), GetShape(weights_), in QuantizedLSTMOpModel()
116 GetShape(biases_), GetShape(prev_cell_state_)}); in QuantizedLSTMOpModel()
118 BuildInterpreter({GetShape(input_), GetShape(prev_output_), in QuantizedLSTMOpModel()
119 GetShape(weights_), GetShape(biases_), in QuantizedLSTMOpModel()
120 GetShape(prev_cell_state_)}); in QuantizedLSTMOpModel()
Dsplit_v_test.cc51 {GetShape(input_), GetShape(size_splits_), GetShape(axis_)}); in SplitVOpModel()
53 BuildInterpreter({GetShape(input_), GetShape(size_splits_), {}}); in SplitVOpModel()
Dreshape_test_common.h91 this->BuildInterpreter({this->GetShape(input_)}); in BuildWithHardcodedShape()
108 {this->GetShape(input_), this->GetShape(shape_input_tensor)}); in BuildWithTensorShape()
127 this->BuildInterpreter({this->GetShape(input_)}); in BuildWithConstantTensorShape()
Dconv3d_test.cc49 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in Conv3dOpModel()
68 BuildInterpreter({GetShape(input_), GetShape(filter_)}); in Conv3dOpModel()
Ddepthwise_conv_hybrid_test.cc62 int bias_size = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel()
102 int input_depth = GetShape(input_)[3]; in BaseDepthwiseConvolutionOpModel()
103 int output_depth = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel()
117 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in BaseDepthwiseConvolutionOpModel()
/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_test.cc74 ASSERT_THAT(GetShape(8), ElementsAre(2, 1)); in TEST_F()
93 ASSERT_THAT(GetShape(2), ElementsAre(2, 2)); in TEST_F()
112 ASSERT_THAT(GetShape(2), ElementsAre(2, 2)); in TEST_F()
135 ASSERT_THAT(GetShape(8), ElementsAre(2, 1)); in TEST_F()
158 ASSERT_THAT(GetShape(9), ElementsAre(1)); in TEST_F()
176 ASSERT_THAT(GetShape(2), ElementsAre(2, 2, 1)); in TEST_F()
194 ASSERT_THAT(GetShape(2), ElementsAre(2, 2, 1)); in TEST_F()
204 ASSERT_THAT(GetShape(2), ElementsAre(2, 2, 1)); in TEST_F()
245 EXPECT_THAT(GetShape(8), ElementsAre(2, 1)); in TEST_F()
253 EXPECT_THAT(GetShape(9), ElementsAre(1)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_expression_test.cc93 TEST_F(XlaExpressionTest, GetShape) { in TEST_F() argument
94 EXPECT_FALSE(XlaExpression().GetShape().ok()); in TEST_F()
95 EXPECT_FALSE(XlaExpression::Invalid().GetShape().ok()); in TEST_F()
98 XlaExpression::Resource(resource_.get()).GetShape()); in TEST_F()
102 XlaExpression::XlaOp(op_, DT_INT32).GetShape()); in TEST_F()
106 XlaExpression::Constant(constant_).GetShape()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing.cc48 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in SliceInMinorDims()
75 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in UpdateSlice()
94 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in UpdateSliceInMinorDims()
118 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in PrependZerosInMajorDims()
134 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in DynamicSliceInMinorDims()
161 TF_ASSIGN_OR_RETURN(Shape index_shape, builder->GetShape(index)); in TorchGather()
162 TF_ASSIGN_OR_RETURN(Shape input_shape, builder->GetShape(input)); in TorchGather()
230 TF_ASSIGN_OR_RETURN(Shape index_shape, builder->GetShape(index)); in TorchScatterDense()
231 TF_ASSIGN_OR_RETURN(Shape input_shape, builder->GetShape(input)); in TorchScatterDense()
266 TF_ASSIGN_OR_RETURN(Shape input_shape, builder->GetShape(input)); in TorchIndexSelect()
[all …]
Dmatrix.cc60 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in GetDiagonalMask()
78 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in GetMatrixDiagonal()
112 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in GetMatrixDiagonalViaGather()
175 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(matrix)); in SetMatrixDiagonal()
176 TF_ASSIGN_OR_RETURN(Shape diag_shape, builder->GetShape(diag)); in SetMatrixDiagonal()
214 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in TriangleMask()
266 TF_ASSIGN_OR_RETURN(Shape x_shape, builder->GetShape(x)); in EinsumDiagonalMask()
292 TF_ASSIGN_OR_RETURN(Shape x_shape, builder->GetShape(x)); in EinsumDiagonal()
307 TF_ASSIGN_OR_RETURN(Shape x_shape, builder->GetShape(x)); in EinsumInverseDiagonal()
376 TF_ASSIGN_OR_RETURN(Shape x_shape, builder->GetShape(x)); in Einsum()
[all …]
Dmath.cc70 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(operand)); in DoWithUpcastToF32()
89 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(operand)); in EnsureOperandIsRealFp()
103 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(operand)); in IsPosInf()
114 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(operand)); in IsNegInf()
141 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(operand)); in IsNegZero()
282 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(x)); in Erfc()
323 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(x)); in Erf()
386 TF_ASSIGN_OR_RETURN(Shape shape, b.GetShape(x)); in ErfInv32()
466 TF_ASSIGN_OR_RETURN(Shape shape, b.GetShape(x)); in ErfInv64()
478 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(x)); in ErfInv()
[all …]
Dtridiagonal.cc57 builder->GetShape(lower_diagonal)); in CheckSystemAndReturnNumEquations()
59 builder->GetShape(main_diagonal)); in CheckSystemAndReturnNumEquations()
61 builder->GetShape(upper_diagonal)); in CheckSystemAndReturnNumEquations()
62 TF_ASSIGN_OR_RETURN(Shape rhs_shape, builder->GetShape(rhs)); in CheckSystemAndReturnNumEquations()
291 TF_ASSIGN_OR_RETURN(Shape diagonals_shape, builder->GetShape(diagonals)); in ThomasSolver()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_utils.cc116 TF_ASSIGN_OR_RETURN(xla::Shape list_shape, list.builder()->GetShape(list)); in IsTensorListInitialized()
127 TF_ASSIGN_OR_RETURN(xla::Shape list_shape, list.builder()->GetShape(list)); in IsNestedTensorList()
145 TF_ASSIGN_OR_RETURN(xla::Shape list_shape, list.builder()->GetShape(list)); in GetTensorListBufferShape()
166 TF_ASSIGN_OR_RETURN(xla::Shape list_shape, list.builder()->GetShape(list)); in GetTensorListPushIndex()
179 TF_ASSIGN_OR_RETURN(xla::Shape list_shape, list.builder()->GetShape(list)); in SetTensorListPushIndex()
210 TF_ASSIGN_OR_RETURN(xla::Shape list_shape, list.builder()->GetShape(list)); in GetLeadingDimForTensorList()
307 TF_ASSIGN_OR_RETURN(xla::Shape element_shape, b->GetShape(element)); in GetInitializedTensorListForElement()
320 TF_ASSIGN_OR_RETURN(xla::Shape original_list_shape, b->GetShape(list)); in GetInitializedTensorListForElement()
362 TF_ASSIGN_OR_RETURN(xla::Shape list_shape, b->GetShape(list)); in ExecuteTensorListPushBack()
369 TF_ASSIGN_OR_RETURN(xla::Shape element_shape, b->GetShape(element)); in ExecuteTensorListPushBack()
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dconv_test.cc41 int bias_size = GetShape(filter_)[0]; in QuantizedConvolutionOpModel()
43 bias_size = GetShape(filter_)[3]; in QuantizedConvolutionOpModel()
77 int input_depth = GetShape(input_)[3]; in QuantizedConvolutionOpModel()
78 int output_depth = GetShape(filter_)[3]; in QuantizedConvolutionOpModel()
95 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in QuantizedConvolutionOpModel()
Dstrided_slice_test.cc43 BuildInterpreter({GetShape(input_), GetShape(begin_), GetShape(end_), in StridedSliceOpModel()
44 GetShape(strides_)}); in StridedSliceOpModel()
Dmin_max_builder_test.cc32 BuildInterpreter({GetShape(input1_), GetShape(input2_)}); in MinMaxOpModel()
45 BuildInterpreter({GetShape(input1_), GetShape(input2_)}); in MinMaxOpModel()
Dtranspose_conv_test.cc40 int bias_size = GetShape(filter_)[0]; in QuantizedTransposeConvOpModel()
79 {GetShape(output_shape_), GetShape(filter_), GetShape(input_)}); in QuantizedTransposeConvOpModel()
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dscatter.cc39 TF_ASSIGN_OR_RETURN(xla::Shape buffer_shape, builder->GetShape(buffer)); in XlaScatter()
40 TF_ASSIGN_OR_RETURN(xla::Shape updates_shape, builder->GetShape(updates)); in XlaScatter()
41 TF_ASSIGN_OR_RETURN(xla::Shape indices_shape, builder->GetShape(indices)); in XlaScatter()
158 TF_ASSIGN_OR_RETURN(updates_shape, builder->GetShape(new_updates)); in XlaScatter()
Dbroadcast.cc34 TF_ASSIGN_OR_RETURN(xla::Shape input_shape, builder->GetShape(input)); in BroadcastTo()
95 TF_ASSIGN_OR_RETURN(auto lhs_xla_shape, lhs->builder()->GetShape(*lhs)); in BroadcastOpsToSame()
96 TF_ASSIGN_OR_RETURN(auto rhs_xla_shape, rhs->builder()->GetShape(*rhs)); in BroadcastOpsToSame()
/external/llvm-project/flang/include/flang/Evaluate/
Dshape.h57 std::optional<Shape> GetShape(FoldingContext &, const A &);
142 if (std::optional<Shape> xShape{GetShape(context_, x)}) { in GetArrayConstructorValueExtent()
185 std::optional<Shape> GetShape(FoldingContext &context, const A &x) { in GetShape() function
192 if (auto shape{GetShape(context, x)}) { in GetConstantShape()
202 if (auto shape{GetShape(context, x)}) { in GetConstantExtents()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dsort_util.cc195 IrArray::Index keys_index(keys_multi_index, params[i].GetShape(), in EmitTiledCompareLoop()
274 IrArray::Index keys_index(keys_multi_index, params[i].GetShape(), in EmitTiledCompareLoop()
309 const Shape& keys_shape = values_arrays[0].GetShape(); in EmitSortInPlace()
334 values_arrays[i].GetShape().element_type(), module), in EmitSortInPlace()
362 IrArray::Index keys_index(keys_multi_index, values_arrays[0].GetShape(), in EmitSortInPlace()
372 values_arrays[operand].GetShape(), in EmitSortInPlace()
380 values_arrays[operand].GetShape(), in EmitSortInPlace()
Dloop_emitter.cc55 shape_(target_array.GetShape()),
84 shape_(target_arrays[0].GetShape()), in LoopEmitter()
89 CHECK(ShapeUtil::SameDimensions(shape_, array.GetShape())) in LoopEmitter()
91 << array.GetShape().ShortDebugString() << "'"; in LoopEmitter()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_nnapi_failure_handling_test.cc101 BuildInterpreter({GetShape(input1_), GetShape(input2_), GetShape(input3_)}, in Init()
Dnnapi_delegate_device_selection_test.cc56 BuildInterpreter({GetShape(input1_), GetShape(input2_)}, /*num_threads=*/-1, in Init()
415 BuildInterpreter({GetShape(input1_), GetShape(input2_), GetShape(input3_)}, in Init()
598 BuildInterpreter({GetShape(input1_), GetShape(input2_)}, /*num_threads=*/-1, in Init()
716 int bias_size = GetShape(filter_)[0];
728 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)},
877 BuildInterpreter({GetShape(input_), GetShape(zero_input_)}, in Init()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc43 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(a)); in DiagonalBlocks()
114 TF_ASSIGN_OR_RETURN(Shape blocks_shape, builder->GetShape(last_blocks)); in DiagonalBlocks()
140 TF_ASSIGN_OR_RETURN(Shape blocks_shape, builder->GetShape(inv_diag_blocks)); in SolveWithInvertedDiagonalBlocks()
141 TF_ASSIGN_OR_RETURN(Shape b_shape, builder->GetShape(b)); in SolveWithInvertedDiagonalBlocks()
144 TF_ASSIGN_OR_RETURN(Shape a_shape, builder->GetShape(a)); in SolveWithInvertedDiagonalBlocks()
251 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(diag_blocks)); in InvertDiagonalBlocks()
378 TF_ASSIGN_OR_RETURN(Shape a_shape, builder->GetShape(a)); in SolveByInvertingDiagonalBlocks()
422 TF_ASSIGN_OR_RETURN(Shape a_shape, builder->GetShape(a)); in SolveDirectly()
423 TF_ASSIGN_OR_RETURN(Shape b_shape, builder->GetShape(b)); in SolveDirectly()
479 TF_ASSIGN_OR_RETURN(Shape a_shape, builder->GetShape(a)); in BuildTriangularSolve()
[all …]

123456789