Searched refs:size_values (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | slice.py | 136 size_values = np.array(parameters["size"]).astype(index_type)
137 out = tf.slice(input_tensor, begin_values, size_values)
165 size_values = np.array(parameters["size"]).astype(index_type)
166 values = [input_values, begin_values, size_values]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 132 ArrayRef<int64_t> size_values, in SliceRankedTensor() argument
140 (begin_values[i] + size_values[i] > input_shape[i])) { in SliceRankedTensor()
151 CreateI64DenseConst(builder, size_shape, size_values, location); in SliceRankedTensor()
156 size_values, in SliceRankedTensor()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 282 std::vector<int64_t> size_values; in matchAndRewrite() local
285 size_values.emplace_back(std::get<1>(pair).getSExtValue() - in matchAndRewrite()
290 RankedTensorType::get({static_cast<int64_t>(size_values.size())}, in matchAndRewrite()
294 slice_op.getLoc(), DenseIntElementsAttr::get(ty, size_values)); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | array_ops_test.py | 1720 size_values = 47
1725 arr = np.random.uniform(size=[batch_size, size_values]).astype(
1767 size_values = 47
1772 arr = np.random.uniform(size=[batch_size, size_values]).astype(
1815 size_values = 47
1823 size_values]).astype(np.int64)
1865 size_values = 47
1873 size_values]).astype(np.int64)
|