Home
last modified time | relevance | path

Searched defs:shape1 (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes_test.py33 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/llvm-project/mlir/lib/Dialect/
DTraits.cpp16 bool OpTrait::util::staticallyKnownBroadcastable(ArrayRef<int64_t> shape1, in staticallyKnownBroadcastable()
33 bool OpTrait::util::getBroadcastedShape(ArrayRef<int64_t> shape1, in getBroadcastedShape()
169 static bool areCompatibleShapes(ArrayRef<int64_t> shape1, in areCompatibleShapes()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dprocess_broadcast_shapes.h39 const RuntimeShape& shape1, in ProcessBroadcastShapes()
/external/tensorflow/tensorflow/python/keras/applications/
Dapplications_load_weight_test.py96 def assertShapeEqual(self, shape1, shape2): argument
Dapplications_test.py79 def assertShapeEqual(self, shape1, shape2): argument
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Delementwise_test.cc233 const BHWC shape1(1, 1, 1, 2); in TEST() local
296 const BHWC shape1(1, 1, 1, 2); in TEST() local
360 const BHWC shape1(1, 1, 1, 2); in TEST() local
424 const BHWC shape1(1, 1, 1, 2); in TEST() local
488 const BHWC shape1(1, 1, 1, 2); in TEST() local
551 const BHWC shape1(1, 1, 1, 2); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc89 Shape shape1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
147 Shape shape1 = ShapeUtil::MakeShape(BF16, {3, 2}); in TEST() local
153 Shape shape1 = ShapeUtil::MakeShape(BF16, {3, 2}); in TEST() local
276 Shape shape1 = ShapeUtil::MakeShape(F32, {}); in TEST() local
288 Shape shape1 = ShapeUtil::MakeShape(F32, {5, 7}); in TEST() local
557 Shape shape1 = ShapeUtil::MakeShape(S32, {1, 9, 4, 1}); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/python/
Doutfeed_receiver_test.cc167 const Shape shape1 = ShapeUtil::MakeShape(U32, {128}); in TEST() local
210 const Shape shape1 = ShapeUtil::MakeShape(U32, {128}); in TEST() local
252 const Shape shape1 = ShapeUtil::MakeShape(U32, {128}); in TEST() local
/external/tensorflow/tensorflow/lite/swift/Tests/
DTensorTests.swift95 let shape1 = Tensor.Shape(2, 2, 3) in testEquatable() variable
/external/llvm-project/mlir/lib/IR/
DTypeUtilities.cpp54 LogicalResult mlir::verifyCompatibleShape(ArrayRef<int64_t> shape1, in verifyCompatibleShape()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dshape_optimizer.cc169 const TensorShapeProto& shape1 = prop1[0].shape(); in Optimize() local
/external/tensorflow/tensorflow/core/kernels/
Dset_kernels.cc380 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
Dsegment_reduction_ops_test.cc50 TensorShape shape1({num_rows, num_cols}); in BM_SegmentReduction() local
Dnn_ops_test.cc850 TensorShape shape1({batch_size, rows, cols, depth}); in BM_AvgPool() local
1045 TensorShape shape1({batch_size, rows, cols, depth}); in BM_MaxPool() local
1224 TensorShape shape1({batch_size, rows, cols, depth}); in BM_ReluFloat() local
1295 TensorShape shape1({batch_size, rows, cols, depth}); in BM_SoftplusFloat() local
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.cc41 const flatbuffers::Vector<int32_t>* shape1 = in HaveSameShapes() local
/external/tensorflow/tensorflow/core/ops/
Dio_ops.cc94 ShapeHandle shape0, shape1, shape2; in __anonec23b74b0302() local
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc1033 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
1740 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
1848 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
1966 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
2133 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
2283 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
2444 GraphDefBuilder shape1(GraphDefBuilder::kFailImmediately); in TEST() local
/external/tensorflow/tensorflow/python/keras/layers/
Dmerge.py51 def _compute_elemwise_op_output_shape(self, shape1, shape2): argument
/external/gemmlowp/test/
Dbenchmark_all_sizes.cc216 bool operator<(const Shape& shape1, const Shape& shape2) { in operator <()
/external/tensorflow/tensorflow/python/ops/distributions/
Dutil.py706 def prefer_static_broadcast_shape(shape1, argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover_test.cc493 auto shape1 = ShapeUtil::MakeShape(F32, {1, 8, 1, 7}); in TEST_F() local
Dalgebraic_simplifier_test.cc5985 auto shape1 = ShapeUtil::MakeShape(F32, {2, 6}); in TEST_F() local
6015 auto shape1 = ShapeUtil::MakeShape(F32, {2, 6}); in TEST_F() local
6040 auto shape1 = ShapeUtil::MakeShape(F32, {6, 2}); in TEST_F() local
6065 auto shape1 = ShapeUtil::MakeShape(F32, {2, 8}); in TEST_F() local
6093 auto shape1 = ShapeUtil::MakeShape(F32, {2, 6, 2}); in TEST_F() local
6214 auto shape1 = ShapeUtil::MakeShape(F32, {6, 5}); in TEST_F() 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/XNNPACK/src/xnnpack/
Dsubgraph.h231 struct xnn_shape shape1; member
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcpu_gpu_fusion_test.cc855 Shape shape1 = ShapeUtil::MakeShape(F32, {param1_dim0, param1_dim1}); in BM_ParallelFusion() local

12