Searched defs:shape0 (Results 1 – 6 of 6) 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
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | io_ops.cc | 94 ShapeHandle shape0, shape1, shape2; in __anone43acdb40302() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 532 Shape shape0 = ShapeUtil::MakeShape(S32, {9, 1, 4}); in TEST() local
|
| /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/compiler/xla/tests/ |
| D | fusion_test.cc | 854 Shape shape0 = ShapeUtil::MakeShape(F32, {param0_dim0, param0_dim1}); in BM_ParallelFusion() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 42 inline bool ProcessBroadcastShapes(const RuntimeShape& shape0, in ProcessBroadcastShapes()
|