| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes_test.py | 33 def _assert_incompatible_broadcast(self, shape1, shape2): argument
50 def _assert_broadcast(self, expected, shape1, shape2): argument
132 def _assert_broadcast_with_unknown_dims(self, expected, shape1, shape2): argument
|
| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | applications_load_weight_test.py | 92 def assertShapeEqual(self, shape1, shape2): argument
|
| D | applications_test.py | 76 def assertShapeEqual(self, shape1, shape2): argument
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 88 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
146 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
152 Shape shape2 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST() local
275 Shape shape2 = ShapeUtil::MakeTupleShape( in TEST() local
287 Shape shape2 = ShapeUtil::MakeOpaqueShape(); in TEST() local
556 Shape shape2 = ShapeUtil::MakeShape(S32, {3, 1, 12}); in TEST() local
|
| /external/tensorflow/tensorflow/lite/experimental/swift/Tests/ |
| D | TensorTests.swift | 96 var shape2: Tensor.Shape = [2, 2, 3] in testEquatable() variable
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer.cc | 166 const TensorShapeProto& shape2 = prop2[0].shape(); in Optimize() local
|
| D | constant_folding.cc | 599 BCast::Vec shape2; in MaterializeBroadcastGradientArgs() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_test.cc | 55 TensorShape shape2({num_rows}); in BM_SegmentReduction() local
|
| D | set_kernels.cc | 380 Status CheckShapesMatch(VarDimArray shape1, VarDimArray shape2) { in CheckShapesMatch()
391 Status GroupShapeFromInputs(VarDimArray shape1, VarDimArray shape2, in GroupShapeFromInputs()
435 const auto shape2 = TensorShapeToArray(set2_t.shape()); in ComputeDenseToDense() local
|
| D | nn_ops_test.cc | 931 TensorShape shape2({4}); in BM_AvgPoolBk() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | io_ops.cc | 94 ShapeHandle shape0, shape1, shape2; in __anon2d93d3360302() local
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | merge.py | 51 def _compute_elemwise_op_output_shape(self, shape1, shape2): argument
|
| /external/gemmlowp/test/ |
| D | benchmark_all_sizes.cc | 212 bool operator<(const Shape& shape1, const Shape& shape2) { in operator <()
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 707 shape2, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment_test.cc | 767 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST_P() local
1341 Shape shape2 = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_P() local
1402 Shape shape2 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_P() local
1585 Shape shape2 = ShapeUtil::MakeShape(xla::F32, {3, 3}); in TEST_P() local
1656 Shape shape2 = ShapeUtil::MakeShape(xla::F32, {3, 3}); in TEST_P() local
1722 Shape shape2 = ShapeUtil::MakeShape(xla::F32, {3, 3}); in TEST_P() local
|
| D | reshape_mover_test.cc | 494 auto shape2 = ShapeUtil::MakeShape(F32, {8, 7, 1}); in TEST_F() local
|
| D | algebraic_simplifier_test.cc | 5436 auto shape2 = ShapeUtil::MakeShape(F32, {3, 2, 2}); in TEST_F() local
5466 auto shape2 = ShapeUtil::MakeShape(F32, {2, 3, 2}); in TEST_F() local
5491 auto shape2 = ShapeUtil::MakeShape(F32, {2, 3, 2}); in TEST_F() local
5516 auto shape2 = ShapeUtil::MakeShape(F32, {2, 2, 2, 2}); in TEST_F() local
5544 auto shape2 = ShapeUtil::MakeShape(F32, {2, 3, 2, 2}); in TEST_F() local
5665 auto shape2 = ShapeUtil::MakeShape(F32, {1, 3, 2, 4}); in TEST_F() local
|
| /external/XNNPACK/src/xnnpack/ |
| D | subgraph.h | 137 struct xnn_shape shape2; member
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 858 Shape shape2 = ShapeUtil::MakeShape(F32, {param2_dim0, param2_dim1}); in BM_ParallelFusion() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 1061 GraphDefBuilder shape2(GraphDefBuilder::kFailImmediately); in TEST() local
1973 GraphDefBuilder shape2(GraphDefBuilder::kFailImmediately); in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 433 const RuntimeShape& shape2, int index2) { in MatchingDim()
440 const RuntimeShape& shape2, int index2, Args... args) { in MatchingDim()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 449 def _ShapeLessThanOrEqual(shape1, shape2): argument
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 980 ShapeHandle shape1, ShapeHandle shape2) { in OutputAsUnion()
|