| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | eval_const_tensor.cc | 77 InferenceContext* shape_c, in TryToInferTensorOutputFromShapeNode() argument
82 if (shape_c == nullptr) return OkStatus(); in TryToInferTensorOutputFromShapeNode()
83 if (!shape_c->FullyDefined(shape_c->input(0)) && !allow_partial) in TryToInferTensorOutputFromShapeNode()
85 if (!shape_c->RankKnown(shape_c->input(0))) return OkStatus(); in TryToInferTensorOutputFromShapeNode()
87 int src_rank = shape_c->Rank(shape_c->input(0)); in TryToInferTensorOutputFromShapeNode()
93 if (shape_c->ValueKnown(shape_c->Dim(shape_c->input(0), i))) { in TryToInferTensorOutputFromShapeNode()
94 dimension = shape_c->Value(shape_c->Dim(shape_c->input(0), i)); in TryToInferTensorOutputFromShapeNode()
101 dimension = shape_c->kUnknownDim; in TryToInferTensorOutputFromShapeNode()
108 if (shape_c->ValueKnown(shape_c->Dim(shape_c->input(0), i))) { in TryToInferTensorOutputFromShapeNode()
109 flat(i) = shape_c->Value(shape_c->Dim(shape_c->input(0), i)); in TryToInferTensorOutputFromShapeNode()
[all …]
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | GEMMLowp.cpp | 181 …(const SimpleTensor<T_in> &a, const SimpleTensor<T_in_1> &b, TensorShape shape_c, int32_t a_offset… in gemmlowp_matrix_multiply_core() argument
186 SimpleTensor<T_out> c(shape_c, dt); in gemmlowp_matrix_multiply_core()
235 SimpleTensor<T1> gemmlowp(const SimpleTensor<T2> &a, const SimpleTensor<T3> &b, TensorShape shape_c) in gemmlowp() argument
237 return gemmlowp_matrix_multiply_core<T1, T2, T3>(a, b, shape_c, 0, 0); in gemmlowp()
337 …onst SimpleTensor<int8_t> &a, const SimpleTensor<int8_t> &b, TensorShape shape_c, int32_t a_offset…
338 …st SimpleTensor<uint8_t> &a, const SimpleTensor<uint8_t> &b, TensorShape shape_c, int32_t a_offset…
339 …int8_t, int8_t>(const SimpleTensor<int8_t> &a, const SimpleTensor<int8_t> &b, TensorShape shape_c);
340 …8_t, uint8_t>(const SimpleTensor<uint8_t> &a, const SimpleTensor<uint8_t> &b, TensorShape shape_c);
341 …nt8_t, int8_t>(const SimpleTensor<uint8_t> &a, const SimpleTensor<int8_t> &b, TensorShape shape_c);
|
| D | GEMMLowp.h | 39 …y_core(const SimpleTensor<T2> &a, const SimpleTensor<T3> &b, TensorShape shape_c, int32_t a_offset…
42 SimpleTensor<T1> gemmlowp(const SimpleTensor<T2> &a, const SimpleTensor<T3> &b, TensorShape shape_c…
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_where_op.py | 212 shape_c = ragged_tensor_shape.RaggedTensorDynamicShape.from_tensor(
217 shape_c, ragged_tensor_shape.broadcast_dynamic_shape(shape_x, shape_y))
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | GEMMLowp.cpp | 57 shape_a, shape_b, shape_c, a_offset, b_offset) in DATA_TEST_CASE() argument
62 Tensor c = create_tensor<Tensor>(shape_c, DataType::S32); in DATA_TEST_CASE()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1979 const auto shape_c = properties.GetOutputProperties("c").at(0).shape(); in TEST_F() local
1981 EXPECT_EQ(shape_a.dim_size(), shape_c.dim_size()); in TEST_F()
1983 EXPECT_EQ(shape_a.dim(0).size(), shape_c.dim(0).size()); in TEST_F()
1985 EXPECT_EQ(shape_a.dim(1).size(), shape_c.dim(1).size()); in TEST_F()
2001 EXPECT_EQ(shape_e.dim(0).size(), shape_c.dim(0).size()); in TEST_F()
2002 EXPECT_NE(shape_e.dim(1).size(), shape_c.dim(1).size()); in TEST_F()
2012 EXPECT_EQ(shape_h.dim(0).size(), shape_c.dim(0).size()); in TEST_F()
2013 EXPECT_EQ(shape_h.dim(1).size(), shape_c.dim(1).size()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 208 Shape shape_c = ShapeUtil::MakeShape(F32, {4, 3}, {false, true}); in TEST() local
209 *shape_c.mutable_layout() = Layout({0, 1}); in TEST()
213 EXPECT_TRUE(ShapeUtil::Compatible(shape_a, shape_c)); in TEST()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | GEMMFixture.h | 56 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_c, TensorShape output_shape… in setup() argument
59 … _target = compute_target(shape_a, shape_b, shape_c, output_shape, alpha, beta, data_type); in setup()
86 …nst TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_c, const TensorShap… in compute_target() argument
92 TensorType c = create_tensor<TensorType>(shape_c, data_type, 1); in compute_target()
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_methods_invocations.py | 1176 for shape_a, shape_b, shape_c, broadcasts_input in test_cases:
1180 make_arg(shape_c),
|