| /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 | 75 def assertShapeEqual(self, shape1, shape2): argument
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 90 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
148 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
154 Shape shape2 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST() local
277 Shape shape2 = ShapeUtil::MakeTupleShape( in TEST() local
289 Shape shape2 = ShapeUtil::MakeOpaqueShape(); in TEST() local
558 Shape shape2 = ShapeUtil::MakeShape(S32, {3, 1, 12}); in TEST() local
|
| /external/tensorflow/tensorflow/lite/swift/Tests/ |
| D | TensorTests.swift | 96 var shape2: Tensor.Shape = [2, 2, 3] in testEquatable() variable
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | broadcast_args.cc | 40 const TfLiteTensor* shape2; member
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer.cc | 170 const TensorShapeProto& shape2 = prop2[0].shape(); in Optimize() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | set_kernels.cc | 380 Status CheckShapesMatch(VarDimArray shape1, VarDimArray shape2) { in CheckShapesMatch()
391 Status GroupShapeFromInputs(VarDimArray shape1, VarDimArray shape2, in GroupShapeFromInputs()
436 const auto shape2 = TensorShapeToArray(set2_t.shape()); in ComputeDenseToDense() local
|
| D | segment_reduction_ops_test.cc | 54 TensorShape shape2({num_rows}); in BM_SegmentReduction() local
|
| D | nn_ops_test.cc | 947 TensorShape shape2({4}); in BM_AvgPoolBk() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | io_ops.cc | 94 ShapeHandle shape0, shape1, shape2; in __anon3d897ff70302() local
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | merge.py | 47 def _compute_elemwise_op_output_shape(self, shape1, shape2): argument
|
| /external/gemmlowp/test/ |
| D | benchmark_all_sizes.cc | 216 bool operator<(const Shape& shape1, const Shape& shape2) { in operator <()
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 708 shape2, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover_test.cc | 494 auto shape2 = ShapeUtil::MakeShape(F32, {8, 7, 1}); in TEST_F() local
|
| D | memory_space_assignment_test.cc | 979 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST_P() local
2482 Shape shape2 = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_P() local
2543 Shape shape2 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_P() local
2726 Shape shape2 = ShapeUtil::MakeShape(xla::F32, {3, 3}); in TEST_P() local
2797 Shape shape2 = ShapeUtil::MakeShape(xla::F32, {3, 3}); in TEST_P() local
2864 Shape shape2 = ShapeUtil::MakeShape(xla::F32, {3, 3}); in TEST_P() local
|
| D | algebraic_simplifier_test.cc | 6075 auto shape2 = ShapeUtil::MakeShape(F32, {3, 2, 2}); in TEST_F() local
6105 auto shape2 = ShapeUtil::MakeShape(F32, {2, 3, 2}); in TEST_F() local
6130 auto shape2 = ShapeUtil::MakeShape(F32, {2, 3, 2}); in TEST_F() local
6155 auto shape2 = ShapeUtil::MakeShape(F32, {2, 2, 2, 2}); in TEST_F() local
6183 auto shape2 = ShapeUtil::MakeShape(F32, {2, 3, 2, 2}); in TEST_F() local
6304 auto shape2 = ShapeUtil::MakeShape(F32, {1, 3, 2, 4}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 938 def reduce_with_shapes(shape1, shape2, shape3, dimensions_to_reduce=(1,)): argument
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | WorkloadDataValidation.cpp | 359 unsigned int shape2[] = {1, 1, 3, 2}; in BOOST_AUTO_TEST_CASE() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 857 Shape shape2 = ShapeUtil::MakeShape(F32, {param2_dim0, param2_dim1}); in BM_ParallelFusion() local
|
| /external/XNNPACK/src/xnnpack/ |
| D | subgraph.h | 293 struct xnn_shape shape2; member
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 1052 GraphDefBuilder shape2(GraphDefBuilder::kFailImmediately); in TEST() local
2004 GraphDefBuilder shape2(GraphDefBuilder::kFailImmediately); in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 248 const RuntimeShape& shape2, int index2) { in MatchingDim()
255 const RuntimeShape& shape2, int index2, Args... args) { in MatchingDim()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 463 def _ShapeLessThanOrEqual(shape1, shape2): argument
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 1066 Shape shape2, int version) { in SimpleTwoInputsVersioningTest()
|