| /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/llvm-project/mlir/lib/Dialect/ |
| D | Traits.cpp | 17 ArrayRef<int64_t> shape2) { in staticallyKnownBroadcastable()
34 ArrayRef<int64_t> shape2, in getBroadcastedShape()
170 ArrayRef<int64_t> shape2) { in areCompatibleShapes()
|
| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | applications_load_weight_test.py | 96 def assertShapeEqual(self, shape1, shape2): argument
|
| D | applications_test.py | 79 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/llvm-project/mlir/lib/IR/ |
| D | TypeUtilities.cpp | 55 ArrayRef<int64_t> shape2) { in verifyCompatibleShape()
|
| /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()
435 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/lite/tools/versioning/ |
| D | op_version.cc | 43 const flatbuffers::Vector<int32_t>* shape2 = in HaveSameShapes() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | io_ops.cc | 94 ShapeHandle shape0, shape1, shape2; in __anon52e493120302() local
|
| /external/gemmlowp/test/ |
| D | benchmark_all_sizes.cc | 216 bool operator<(const Shape& shape1, const Shape& shape2) { in operator <()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | merge.py | 51 def _compute_elemwise_op_output_shape(self, shape1, shape2): argument
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 707 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 | 960 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST_P() local
2463 Shape shape2 = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_P() local
2524 Shape shape2 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_P() local
2707 Shape shape2 = ShapeUtil::MakeShape(xla::F32, {3, 3}); in TEST_P() local
2778 Shape shape2 = ShapeUtil::MakeShape(xla::F32, {3, 3}); in TEST_P() local
2845 Shape shape2 = ShapeUtil::MakeShape(xla::F32, {3, 3}); in TEST_P() local
|
| D | algebraic_simplifier_test.cc | 5986 auto shape2 = ShapeUtil::MakeShape(F32, {3, 2, 2}); in TEST_F() local
6016 auto shape2 = ShapeUtil::MakeShape(F32, {2, 3, 2}); in TEST_F() local
6041 auto shape2 = ShapeUtil::MakeShape(F32, {2, 3, 2}); in TEST_F() local
6066 auto shape2 = ShapeUtil::MakeShape(F32, {2, 2, 2, 2}); in TEST_F() local
6094 auto shape2 = ShapeUtil::MakeShape(F32, {2, 3, 2, 2}); in TEST_F() local
6215 auto shape2 = ShapeUtil::MakeShape(F32, {1, 3, 2, 4}); in TEST_F() local
|
| /external/XNNPACK/src/xnnpack/ |
| D | subgraph.h | 232 struct xnn_shape shape2; member
|
| /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/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 1050 GraphDefBuilder shape2(GraphDefBuilder::kFailImmediately); in TEST() local
1982 GraphDefBuilder shape2(GraphDefBuilder::kFailImmediately); in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 471 const RuntimeShape& shape2, int index2) { in MatchingDim()
478 const RuntimeShape& shape2, int index2, Args... args) { in MatchingDim()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 457 def _ShapeLessThanOrEqual(shape1, shape2): argument
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 1066 Shape shape2, int version) { in SimpleTwoInputsVersioningTest()
|