| /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/compiler/xla/ |
| D | shape_util_test.cc | 87 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
145 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
151 Shape shape2 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST() local
253 Shape shape2 = ShapeUtil::MakeTupleShape( in TEST() local
265 Shape shape2 = ShapeUtil::MakeOpaqueShape(); in TEST() local
534 Shape shape2 = ShapeUtil::MakeShape(S32, {3, 1, 12}); in TEST() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer.cc | 111 const TensorShapeProto& shape2 = prop2[0].shape(); in Optimize() local
|
| D | constant_folding.cc | 569 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 | 924 TensorShape shape2({4}); in BM_AvgPoolBk() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | io_ops.cc | 94 ShapeHandle shape0, shape1, shape2; in __anone43acdb40302() local
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | merge.py | 48 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 | 700 shape1, shape2, name="prefer_static_broadcast_shape"): 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
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 411 const RuntimeShape& shape2, int index2) { in MatchingDim()
418 const RuntimeShape& shape2, int index2, Args... args) { in MatchingDim()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 427 const RuntimeShape& shape2, int index2) { in MatchingDim()
434 const RuntimeShape& shape2, int index2, Args... args) { in MatchingDim()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | 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 | 1051 GraphDefBuilder shape2(GraphDefBuilder::kFailImmediately); in TEST() local
1936 GraphDefBuilder shape2(GraphDefBuilder::kFailImmediately); in TEST() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 454 def _ShapeLessThanOrEqual(shape1, shape2): argument
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 931 ShapeHandle shape1, ShapeHandle shape2) { in OutputAsUnion()
|
| /external/v8/src/ |
| D | objects.cc | 17287 StringShape shape2(other); in IsMatch() local
|