| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | process_broadcast_shapes.h | 92 const RuntimeShape* shape_b = in ProcessBroadcastShapes() local
103 while (i >= 0 && shape_a->Dims(i) == shape_b->Dims(i)) { in ProcessBroadcastShapes()
104 params->broadcast_shape[4] *= shape_b->Dims(i); in ProcessBroadcastShapes()
110 params->broadcast_shape[3] *= shape_b->Dims(i); in ProcessBroadcastShapes()
113 while (i >= 0 && shape_a->Dims(i) == shape_b->Dims(i)) { in ProcessBroadcastShapes()
118 while (i >= 0 && shape_b->Dims(i) == 1) { in ProcessBroadcastShapes()
122 while (i >= 0 && shape_a->Dims(i) == shape_b->Dims(i)) { in ProcessBroadcastShapes()
123 params->broadcast_shape[0] *= shape_b->Dims(i); in ProcessBroadcastShapes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | horizontal_input_fusion.cc | 63 const Shape& shape_b) { in CompareShapeDimsFromLeftToRight() argument
64 if (shape_a.rank() != shape_b.rank()) { in CompareShapeDimsFromLeftToRight()
65 return shape_a.rank() < shape_b.rank(); in CompareShapeDimsFromLeftToRight()
68 auto dims_b = shape_b.dimensions(); in CompareShapeDimsFromLeftToRight()
98 Shape shape_b = GetInputShapeForMultiOutputFusion(*b); in FindAndSortFusionCandidates() local
99 if (!ShapeUtil::EqualIgnoringElementType(shape_a, shape_b)) { in FindAndSortFusionCandidates()
102 return CompareShapeDimsFromLeftToRight(shape_a, shape_b); in FindAndSortFusionCandidates()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | math_test.py | 283 shape_b = (7, 5) if tr_b else (5, 7)
285 shape_b = (2,) + shape_b
288 y = random_ops.random_uniform(shape_b)
308 shape_b = (4, 7, 5) if tr_b else (4, 5, 7)
310 shape_b = (2,) + shape_b
313 y = random_ops.random_uniform(shape_b)
331 shape_b = (2, 5, 7) if broadcast_b else (4, 2, 5, 7)
333 shape_b = (2,) + shape_b if stack_b else shape_b
335 y = random_ops.random_uniform(shape_b)
710 shape_b = (2, 4, 3, 5) if stack_b else (4, 3, 5)
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | resource_variable_test.cc | 99 std::vector<int> shape_b = {1}; in TEST() local
101 InitTensor(shape_b, kTfLiteDynamic, 4.0, &tensor_b); in TEST()
140 std::vector<int> shape_b = {2}; in TEST() local
142 InitTensor(shape_b, kTfLiteDynamic, 4.0, &tensor_b); in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass_test.cc | 302 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
304 Output shape = ops::Add(s.WithOpName("shape"), shape_a, shape_b); in TEST()
354 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
356 Output shape = ops::Add(s.WithOpName("shape"), shape_a, shape_b); in TEST()
379 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
381 Output shape = ops::Add(s.WithOpName("shape"), shape_a, shape_b); in TEST()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | saved_model_api_test.cc | 236 const TF_Shape* shape_b = TF_TensorSpecShape(tensor_spec_b); in TEST_P() local
241 EXPECT_EQ(0, TF_ShapeDims(shape_b)); in TEST_P()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1821 const auto shape_b = properties.GetOutputProperties("b").at(0).shape(); in TEST_F() local
1823 EXPECT_EQ(1, shape_b.dim_size()); in TEST_F()
1824 EXPECT_EQ(shape_b.dim_size(), shape_d.dim_size()); in TEST_F()
1825 EXPECT_GE(-2, shape_b.dim(0).size()); in TEST_F()
1826 EXPECT_NE(shape_a.dim(0).size(), shape_b.dim(0).size()); in TEST_F()
1827 EXPECT_EQ(shape_b.dim(0).size(), shape_d.dim(0).size()); in TEST_F()
1893 const auto shape_b = properties.GetOutputProperties("b").at(0).shape(); in TEST_F() local
1897 EXPECT_EQ(shape_b.DebugString(), shape_o2.DebugString()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 215 Shape shape_b = ShapeUtil::MakeShape(F32, {4, 3}, {true, false}); in TEST() local
216 *shape_b.mutable_layout() = Layout({0, 1}); in TEST()
221 EXPECT_TRUE(ShapeUtil::Compatible(shape_a, shape_b)); in TEST()
|