Home
last modified time | relevance | path

Searched defs:shape0 (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Delementwise_test.cc232 const BHWC shape0(1, 2, 1, 2); in TEST() local
247 const BHWC shape0(1, 2, 1, 2); in TEST() local
261 const BHWC shape0(1, 2, 1, 2); in TEST() local
295 const BHWC shape0(1, 2, 1, 2); in TEST() local
325 const BHWC shape0(1, 2, 1, 2); in TEST() local
359 const BHWC shape0(1, 2, 1, 2); in TEST() local
389 const BHWC shape0(1, 2, 1, 2); in TEST() local
423 const BHWC shape0(1, 2, 1, 2); in TEST() local
453 const BHWC shape0(1, 2, 1, 2); in TEST() local
487 const BHWC shape0(1, 2, 1, 2); in TEST() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/python/
Doutfeed_receiver_test.cc124 const Shape shape0 = ShapeUtil::MakeShape(U32, {16}); in TEST() local
157 const Shape shape0 = ShapeUtil::MakeShape(U32, {16}); in TEST() local
202 const Shape shape0 = ShapeUtil::MakeShape(U32, {16}); in TEST() local
245 const Shape shape0 = ShapeUtil::MakeShape(U32, {16}); in TEST() local
278 const Shape shape0 = ShapeUtil::MakeShape(U32, {16}); in TEST() local
305 const Shape shape0 = ShapeUtil::MakeShape(U32, {16}); in TEST() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dprocess_broadcast_shapes.h38 inline bool ProcessBroadcastShapes(const RuntimeShape& shape0, in ProcessBroadcastShapes()
/external/tensorflow/tensorflow/core/ops/
Dio_ops.cc94 ShapeHandle shape0, shape1, shape2; in __anon52e493120302() local
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc556 Shape shape0 = ShapeUtil::MakeShape(S32, {9, 1, 4}); in TEST() local
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc689 bool ShapesAgreeUpToBroadcasting(const Shape& shape0, const Shape& shape1) { in ShapesAgreeUpToBroadcasting()
717 bool ShapesAgreeUpToExtending(const Shape& shape0, const Shape& shape1) { in ShapesAgreeUpToExtending()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcpu_gpu_fusion_test.cc853 Shape shape0 = ShapeUtil::MakeShape(F32, {param0_dim0, param0_dim1}); in BM_ParallelFusion() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc1119 BHWC shape0; in Parse() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc1394 SmallVector<int32_t, 4> shape0; in fold() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc1133 Output shape0 = ops::Const(s.WithOpName("shape0"), 4, {}); in TEST_F() local