Home
last modified time | relevance | path

Searched refs:shape_value (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dunfuse_batch_norm.cc37 Value value_1d, Value shape_value, in BroadcastToFeatureDim() argument
43 if (shape_value) { in BroadcastToFeatureDim()
45 loc, result_type, value_1d, shape_value, dims); in BroadcastToFeatureDim()
100 Value shape_value = CalculateShapeValue(op->getLoc(), variance, rewriter); in MaterializeEpsilon() local
102 op->getLoc(), broadcast_to_type, epsilon, shape_value, in MaterializeEpsilon()
141 Value shape_value; in matchAndRewrite() local
143 shape_value = in matchAndRewrite()
148 shape_value, feature_dim, rewriter); in matchAndRewrite()
151 shape_value, feature_dim, rewriter); in matchAndRewrite()
154 shape_value, feature_dim, rewriter); in matchAndRewrite()
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_serialization_ops_test.py258 shape_value = np.array([4, 5], dtype=np.int64)
266 sparse_tensor.dense_shape: shape_value
287 shape_value = np.array([4, 5], dtype=np.int64)
296 sparse_tensor.dense_shape: shape_value
300 self.assertAllEqual(deserialized_value.dense_shape, shape_value)
323 shape_value = np.array([], dtype=np.int64)
334 sparse_tensor.dense_shape: shape_value
338 self.assertAllEqual(deserialized_value.dense_shape, shape_value)
345 shape_value = np.array([], dtype=np.int64)
356 sparse_tensor.dense_shape: shape_value
Dsparse_tensors_map_ops_test.py137 shape_value = np.array([4, 5], dtype=np.int64)
147 sparse_tensor.dense_shape: shape_value
152 self.assertAllEqual(roundtrip_value.dense_shape, shape_value)
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1076 DenseIntElementsAttr shape_value; in Verify() local
1077 if (matchPattern(shape, m_Constant(&shape_value))) { in Verify()
1078 for (const auto shape_elem : shape_value) { in Verify()
1086 for (auto shape_it : llvm::enumerate(shape_value)) { in Verify()
1102 for (auto shape_it : llvm::enumerate(shape_value)) { in Verify()
1108 << "] must be equal to the value of shape " << shape_value; in Verify()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_list_ops_decomposition.cc535 LogicalResult GetConstShapeValue(Value shape_value, in GetConstShapeValue() argument
537 auto shape_op = shape_value.getDefiningOp(); in GetConstShapeValue()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py3608 shape_value = tensor_util.constant_value(shape)
3611 if shape_value is not None:
3612 shape_value = np.asarray(shape_value)
3613 if -1 in shape_value:
3615 elif not shape_value.size:
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc2877 TensorValue shape_value(&tensor); in SimplifyReduction() local
2879 if (!CreateNodeDef(OptimizedNodeName(*node, "_shape_const"), shape_value, in SimplifyReduction()