| /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 | 86 Shape shape1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
144 Shape shape1 = ShapeUtil::MakeShape(BF16, {3, 2}); in TEST() local
150 Shape shape1 = ShapeUtil::MakeShape(BF16, {3, 2}); in TEST() local
252 Shape shape1 = ShapeUtil::MakeShape(F32, {}); in TEST() local
264 Shape shape1 = ShapeUtil::MakeShape(F32, {5, 7}); in TEST() local
533 Shape shape1 = ShapeUtil::MakeShape(S32, {1, 9, 4, 1}); in TEST() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer.cc | 110 const TensorShapeProto& shape1 = prop1[0].shape(); in Optimize() local
|
| D | constant_folding.cc | 565 BCast::Vec shape1; in MaterializeBroadcastGradientArgs() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_debug.cc | 100 const xla::Shape& shape1 = in TFE_TensorHandleTensorDebugInfo() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_test.cc | 51 TensorShape shape1({num_rows, num_cols}); 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()
434 const auto shape1 = TensorShapeToArray(set1_t.shape()); in ComputeDenseToDense() local
|
| D | nn_ops_test.cc | 825 TensorShape shape1({batch_size, rows, cols, depth}); in BM_AvgPool() local
1021 TensorShape shape1({batch_size, rows, cols, depth}); in BM_MaxPool() local
1201 TensorShape shape1({batch_size, rows, cols, depth}); in BM_ReluFloat() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | io_ops.cc | 94 ShapeHandle shape0, shape1, shape2; in __anone43acdb40302() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 1034 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
1704 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
1807 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
1920 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
2075 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
2214 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
2360 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() 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 | 493 auto shape1 = ShapeUtil::MakeShape(F32, {1, 8, 1, 7}); in TEST_F() local
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 410 inline int MatchingDim(const RuntimeShape& shape1, int index1, in MatchingDim()
417 int MatchingDim(const RuntimeShape& shape1, int index1, in MatchingDim()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 661 bool ShapesAgreeUpToBroadcasting(const Shape& shape0, const Shape& shape1) { in ShapesAgreeUpToBroadcasting()
689 bool ShapesAgreeUpToExtending(const Shape& shape0, const Shape& shape1) { in ShapesAgreeUpToExtending()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 426 inline int MatchingDim(const RuntimeShape& shape1, int index1, in MatchingDim()
433 int MatchingDim(const RuntimeShape& shape1, int index1, in MatchingDim()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 856 Shape shape1 = ShapeUtil::MakeShape(F32, {param1_dim0, param1_dim1}); in BM_ParallelFusion() 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/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 43 const RuntimeShape& shape1, in ProcessBroadcastShapes()
|
| /external/v8/src/ |
| D | objects.cc | 17286 StringShape shape1(string_); in IsMatch() local
|