Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_test.cc71 ASSERT_THAT(GetShape(8), ElementsAre(2, 1)); in TEST_F()
90 ASSERT_THAT(GetShape(2), ElementsAre(2, 2)); in TEST_F()
109 ASSERT_THAT(GetShape(2), ElementsAre(2, 2)); in TEST_F()
132 ASSERT_THAT(GetShape(8), ElementsAre(2, 1)); in TEST_F()
155 ASSERT_THAT(GetShape(9), ElementsAre(1)); in TEST_F()
173 ASSERT_THAT(GetShape(2), ElementsAre(2, 2, 1)); in TEST_F()
191 ASSERT_THAT(GetShape(2), ElementsAre(2, 2, 1)); in TEST_F()
201 ASSERT_THAT(GetShape(2), ElementsAre(2, 2, 1)); in TEST_F()
242 EXPECT_THAT(GetShape(8), ElementsAre(2, 1)); in TEST_F()
250 EXPECT_THAT(GetShape(9), ElementsAre(1)); in TEST_F()
[all …]
Dkernel_test.cc124 ASSERT_THAT(GetShape(8), ElementsAre(2, 1)); in TEST_F()
135 ASSERT_THAT(GetShape(8), ElementsAre(3, 1)); in TEST_F()
221 ASSERT_THAT(GetShape(8), ElementsAre(2, 1)); in TEST_F()
270 ASSERT_THAT(GetShape(17), ElementsAre(1)); in TEST_F()
286 ASSERT_THAT(GetShape(17), ElementsAre(1)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing.cc28 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in SliceInMinorDims()
55 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in UpdateSlice()
73 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in UpdateSliceInMinorDims()
97 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in PrependZerosInMajorDims()
113 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in DynamicSliceInMinorDims()
140 TF_ASSIGN_OR_RETURN(Shape index_shape, builder->GetShape(index)); in TorchGather()
143 TF_ASSIGN_OR_RETURN(Shape input_shape, builder->GetShape(input)); in TorchGather()
167 TF_ASSIGN_OR_RETURN(Shape input_shape, builder->GetShape(input)); in TorchIndexSelect()
168 TF_ASSIGN_OR_RETURN(Shape index_shape, builder->GetShape(index)); in TorchIndexSelect()
Dmath.cc38 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(operand)); in DoWithUpcastToF32()
57 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(operand)); in EnsureOperandIsRealFp()
71 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(operand)); in IsPosInf()
82 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(operand)); in IsNegInf()
109 TF_ASSIGN_OR_RETURN(auto shape, b.GetShape(operand)); in IsNegZero()
284 TF_ASSIGN_OR_RETURN(Shape shape, b.GetShape(x)); in ErfInv()
557 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(x)); in MaybeConjugate()
567 TF_ASSIGN_OR_RETURN(auto shape, builder->GetShape(from)); in NextAfter()
Dpooling.cc79 TF_ASSIGN_OR_RETURN(Shape operand_shape, b->GetShape(operand)); in ComputeSums()
80 TF_ASSIGN_OR_RETURN(Shape init_shape, b->GetShape(init_value)); in ComputeSums()
138 TF_ASSIGN_OR_RETURN(Shape operand_shape, b->GetShape(operand)); in MaxPool()
154 TF_ASSIGN_OR_RETURN(Shape operand_shape, b->GetShape(operand)); in AvgPool()
208 b->GetShape(out_backprop)); in AvgPoolGrad()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc90 TF_ASSIGN_OR_RETURN(xla::Shape shape, builder->GetShape(x)); in operator >>()
104 StatusOr<Shape> XlaBuilder::GetShape(const XlaOp& op) const { in GetShape() function in xla::XlaBuilder
115 TF_ASSIGN_OR_RETURN(const Shape& shape, GetShape(operand)); in GetOperandShapes()
435 TF_ASSIGN_OR_RETURN(const Shape& operand_shape, GetShape(operand)); in AddBroadcastSequence()
474 TF_ASSIGN_OR_RETURN(const Shape& operand_shape, GetShape(operand)); in UnaryOp()
487 TF_ASSIGN_OR_RETURN(const Shape& lhs_shape, GetShape(lhs)); in BinaryOp()
488 TF_ASSIGN_OR_RETURN(const Shape& rhs_shape, GetShape(rhs)); in BinaryOp()
538 TF_ASSIGN_OR_RETURN(Shape updated_lhs_shape, GetShape(updated_lhs)); in BinaryOp()
543 TF_ASSIGN_OR_RETURN(Shape updated_rhs_shape, GetShape(updated_rhs)); in BinaryOp()
557 TF_ASSIGN_OR_RETURN(const Shape& lhs_shape, GetShape(lhs)); in TernaryOp()
[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()
Dxla_expression.cc127 TF_ASSIGN_OR_RETURN(TensorShape shape, GetShape()); in ResolveConstant()
141 xla::StatusOr<TensorShape> XlaExpression::GetShape() const { in GetShape() function in tensorflow::XlaExpression
147 handle().builder()->GetShape(handle())); in GetShape()
/external/tensorflow/tensorflow/lite/kernels/
Dsplit_v_test.cc47 {GetShape(input_), GetShape(size_splits_), GetShape(axis_)}); in SplitVOpModel()
49 BuildInterpreter({GetShape(input_), GetShape(size_splits_), {}}); in SplitVOpModel()
Ddepthwise_conv_test.cc52 int bias_size = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel()
92 int input_depth = GetShape(input_)[3]; in BaseDepthwiseConvolutionOpModel()
93 int output_depth = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel()
107 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in BaseDepthwiseConvolutionOpModel()
Dreshape_test.cc88 BuildInterpreter({GetShape(input_)}); in BuildWithHardcodedShape()
102 BuildInterpreter({GetShape(input_), GetShape(shape_input_tensor)}); in BuildWithTensorShape()
119 BuildInterpreter({GetShape(input_)}); in BuildWithConstantTensorShape()
Dsplit_test.cc44 BuildInterpreter({GetShape(axis_), GetShape(input_)}); in SplitOpModel()
46 BuildInterpreter({{}, GetShape(input_)}); in SplitOpModel()
Drange_test.cc36 BuildInterpreter({GetShape(start_), GetShape(limit_), GetShape(delta_)}); in RangeOpModel()
Dtranspose_conv_test.cc60 {GetShape(output_shape_), GetShape(filter_), GetShape(input_)}); in TransposeConvOpModel()
Ddetection_postprocess_test.cc70 BuildInterpreter({GetShape(input1_), GetShape(input2_), GetShape(input3_)}); in BaseDetectionPostprocessOpModel()
281 BuildInterpreter({GetShape(input1_), GetShape(input2_), GetShape(input3_)}); in DetectionPostprocessOpModelwithRegularNMS()
Dmfcc_test.cc56 BuildInterpreter({GetShape(input1_), GetShape(input2_)}); in BaseMfccOpModel()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc377 const Shape& lhs_shape = lhs_array_.GetShape(); in Emit()
378 const Shape& rhs_shape = rhs_array_.GetShape(); in Emit()
409 const Shape& lhs_shape = lhs_array_.GetShape(); in EmitNaiveLlvmIrGemm()
410 const Shape& rhs_shape = rhs_array_.GetShape(); in EmitNaiveLlvmIrGemm()
538 target_multi_index, target_array_.GetShape(), lhs_index.GetType()); in EmitNaiveLlvmIrGemm()
556 if (ShapeUtil::ElementIsComplex(lhs_array_.GetShape())) { in EmitScalarDot()
592 PrimitiveType type = target_array_.GetShape().element_type(); in EmitCallToRuntime()
687 const Shape& lhs_shape = lhs_array_.GetShape(); in GetMatMultDims()
688 const Shape& rhs_shape = rhs_array_.GetShape(); in GetMatMultDims()
700 LayoutUtil::Minor(target_array_.GetShape().layout(), 0) == 0}; in GetMatMultDims()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dloop_emitter.cc47 shape_(target_array.GetShape()),
76 shape_(target_arrays[0].GetShape()), in LoopEmitter()
81 CHECK(ShapeUtil::SameDimensions(shape_, array.GetShape())) in LoopEmitter()
83 << array.GetShape().ShortDebugString() << "'"; in LoopEmitter()
Dsort_util.cc194 IrArray::Index keys_index(keys_multi_index, params[i].GetShape(), in EmitTiledCompareLoop()
272 IrArray::Index keys_index(keys_multi_index, params[i].GetShape(), in EmitTiledCompareLoop()
307 const Shape& keys_shape = values_arrays[0].GetShape(); in EmitSortInPlace()
332 values_arrays[i].GetShape().element_type(), module), in EmitSortInPlace()
360 IrArray::Index keys_index(keys_multi_index, values_arrays[0].GetShape(), in EmitSortInPlace()
370 values_arrays[operand].GetShape(), in EmitSortInPlace()
378 values_arrays[operand].GetShape(), in EmitSortInPlace()
Ddynamic_update_slice_util.cc46 const Shape& output_shape = output_array.GetShape(); in EmitDynamicUpdateSliceInPlaceImpl()
116 Shape output_shape = output_array.GetShape(); in EmitDynamicUpdateSliceInPlace()
117 Shape update_shape = update_array.GetShape(); in EmitDynamicUpdateSliceInPlace()
127 bool is_signed = ShapeUtil::ElementIsSigned(start_indices_array.GetShape()); in EmitDynamicUpdateSliceInPlace()
/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()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc100 BuildInterpreter({GetShape(input1_), GetShape(input2_)}, in FloatAddOpModel()
171 BuildInterpreter({GetShape(input1_), GetShape(input2_)}); in FloatMulOpModel()
210 BuildInterpreter({GetShape(input_)}); in FloatPoolingOpModel()
275 int bias_size = GetShape(filter_)[0]; in ConvolutionOpModel()
302 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in ConvolutionOpModel()
454 int bias_size = GetShape(filter_)[3]; in DepthwiseConvolutionOpModel()
468 int input_depth = GetShape(input_)[3]; in DepthwiseConvolutionOpModel()
469 int output_depth = GetShape(filter_)[3]; in DepthwiseConvolutionOpModel()
479 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in DepthwiseConvolutionOpModel()
552 BuildInterpreter({GetShape(input_), GetShape(weights_), GetShape(bias_)}); in FullyConnectedOpModel()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc42 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(a)); in DiagonalBlocks()
104 TF_ASSIGN_OR_RETURN(Shape blocks_shape, builder->GetShape(last_blocks)); in DiagonalBlocks()
131 TF_ASSIGN_OR_RETURN(Shape shape, builder->GetShape(diag_blocks)); in InvertDiagonalBlocks()
258 TF_ASSIGN_OR_RETURN(Shape blocks_shape, builder->GetShape(inv_diag_blocks)); in SolveWithInvertedDiagonalBlocks()
259 TF_ASSIGN_OR_RETURN(Shape b_shape, builder->GetShape(b)); in SolveWithInvertedDiagonalBlocks()
262 TF_ASSIGN_OR_RETURN(Shape a_shape, builder->GetShape(a)); in SolveWithInvertedDiagonalBlocks()
356 TF_ASSIGN_OR_RETURN(Shape a_shape, builder->GetShape(a)); in BuildTriangularSolve()
357 TF_ASSIGN_OR_RETURN(Shape b_shape, builder->GetShape(b)); in BuildTriangularSolve()
/external/tensorflow/tensorflow/core/kernels/
Dshape_ops.h32 inline Status GetShape(OpKernelContext* ctx, int input_index, in GetShape() function
46 OP_REQUIRES_OK(ctx, shape_op_helpers::GetShape(ctx, 0, &shape)); in Compute()
74 OP_REQUIRES_OK(ctx, shape_op_helpers::GetShape(ctx, i, &shape)); in Compute()
102 OP_REQUIRES_OK(ctx, shape_op_helpers::GetShape(ctx, 0, &shape)); in Compute()
119 OP_REQUIRES_OK(ctx, shape_op_helpers::GetShape(ctx, 0, &shape)); in Compute()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dwhile_test.cc229 builder.GetShape(result).ConsumeValueOrDie()); in XLA_TEST_F()
283 builder.GetShape(result).ConsumeValueOrDie()); in XLA_TEST_F()
343 builder.GetShape(result).ConsumeValueOrDie()); in XLA_TEST_F()
398 builder.GetShape(result).ConsumeValueOrDie()); in XLA_TEST_F()
456 builder.GetShape(result).ConsumeValueOrDie()); in XLA_TEST_F()
507 builder.GetShape(result).ConsumeValueOrDie()); in XLA_TEST_F()
554 builder.GetShape(result).ConsumeValueOrDie()); in XLA_TEST_F()
599 builder.GetShape(result).ConsumeValueOrDie()); in XLA_TEST_F()
690 builder.GetShape(while_result2).ConsumeValueOrDie()); in XLA_TEST_F()
694 builder.GetShape(result).ConsumeValueOrDie()); in XLA_TEST_F()
[all …]

123456