Searched refs:shape0 (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_debug.cc | 99 xla::Shape shape0 = xla::ShapeUtil::GetTupleElementShape(padded_shape, 0); in TFE_TensorHandleTensorDebugInfo() local
102 if (shape0.IsTuple() || shape1.IsTuple()) { in TFE_TensorHandleTensorDebugInfo()
108 if (!xla::ShapeUtil::Equal(shape0, shape1)) { in TFE_TensorHandleTensorDebugInfo()
119 padded_shape = shape0; in TFE_TensorHandleTensorDebugInfo()
|
| /external/tensorflow/tensorflow/contrib/graph_editor/ |
| D | reroute.py | 63 shape0, shape1 = t0.get_shape(), t1.get_shape()
64 if not shape0.is_compatible_with(shape1):
65 raise ValueError("Shapes {} and {} are not compatible.".format(shape0,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | concat_ops_test.py | 259 for shape0 in (), (2,):
260 axis = len(shape0)
264 x0 = np.random.randn(*(shape0 + (n0,) + shape1))
265 x1 = np.random.randn(*(shape0 + (n1,) + shape1))
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | io_ops.cc | 94 ShapeHandle shape0, shape1, shape2; in __anone43acdb40302() local
95 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &shape0)); in __anone43acdb40302()
98 TF_RETURN_IF_ERROR(c->Merge(shape1, shape2, &shape0)); in __anone43acdb40302()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 661 bool ShapesAgreeUpToBroadcasting(const Shape& shape0, const Shape& shape1) { in ShapesAgreeUpToBroadcasting() argument
662 CheckNonEmptyShapeDimensions(shape0); in ShapesAgreeUpToBroadcasting()
665 const Shape* longer = &shape0; in ShapesAgreeUpToBroadcasting()
667 if (shape1.dimensions_count() > shape0.dimensions_count()) { in ShapesAgreeUpToBroadcasting()
669 shorter = &shape0; in ShapesAgreeUpToBroadcasting()
689 bool ShapesAgreeUpToExtending(const Shape& shape0, const Shape& shape1) { in ShapesAgreeUpToExtending() argument
690 CheckNonEmptyShapeDimensions(shape0); in ShapesAgreeUpToExtending()
693 const Shape* longer = &shape0; in ShapesAgreeUpToExtending()
695 if (shape1.dimensions_count() > shape0.dimensions_count()) { in ShapesAgreeUpToExtending()
697 shorter = &shape0; in ShapesAgreeUpToExtending()
|
| D | tooling_util.h | 130 bool ShapesAgreeUpToBroadcasting(const Shape& shape0, const Shape& shape1);
141 bool ShapesAgreeUpToExtending(const Shape& shape0, const Shape& shape1);
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | concat_op_test.py | 375 for shape0 in (), (2,):
376 axis = len(shape0)
380 x0 = np.random.randn(*(shape0 + (n0,) + shape1))
381 x1 = np.random.randn(*(shape0 + (n1,) + shape1))
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 532 Shape shape0 = ShapeUtil::MakeShape(S32, {9, 1, 4}); in TEST() local
536 ShapeUtil::InsertedOrDeleted1SizedDimensions(shape0, shape1))); in TEST()
538 ShapeUtil::InsertedOrDeleted1SizedDimensions(shape0, shape2))); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 854 Shape shape0 = ShapeUtil::MakeShape(F32, {param0_dim0, param0_dim1}); in BM_ParallelFusion() local
855 auto param0 = Parameter(&builder, 0, shape0, "param0"); in BM_ParallelFusion()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 42 inline bool ProcessBroadcastShapes(const RuntimeShape& shape0, in ProcessBroadcastShapes() argument
46 std::max(shape0.DimensionsCount(), shape1.DimensionsCount()); in ProcessBroadcastShapes()
51 auto extended_shape0 = RuntimeShape::ExtendedShape(dims_count, shape0); in ProcessBroadcastShapes()
|