- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for begin_values (0.29 sec)
-
tensorflow/compiler/mlir/lite/utils/utils.h
inline TypedAttr GetOffSet(Attribute begin, Attribute end) { auto begin_values = begin.cast<DenseElementsAttr>().getValues<int32_t>(); auto end_values = end.cast<DenseElementsAttr>().getValues<int32_t>(); SmallVector<int32_t> offsets; if (begin_values.size() == end_values.size()) { for (size_t i = 0; i < begin_values.size(); ++i) { offsets.push_back(end_values[i] - begin_values[i]); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
for (int i = 0, end = input_shape.size(); i < end; i++) { if (begin_values[i] < 0 || (begin_values[i] + size_values[i] > input_shape[i])) { return CreateF32SplatConst(builder, size_shape, 0, location); } } // Create a dense constant op for slice's begin auto slice_i2c_begin = CreateI64DenseConst(builder, begin_shape, begin_values, location); // Create a dense constant op for slice's size
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
auto create_slice_op = [&](int32_t axis_i, int32_t begin_i, int32_t size_i, Value input) { SmallVector<int64_t, 4> begin_values(input_rank, 0); begin_values[axis_i] = begin_i; auto begin_attr = DenseIntElementsAttr::get(axis_type, begin_values); auto begin = rewriter.create<ConstOp>(op->getLoc(), axis_type, begin_attr); SmallVector<int64_t, 4> output_shape;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
loc, rewriter.getIndexType(), begin_value); begin_values.push_back(begin_value_casted); } auto index_ty = rewriter.getIndexType(); auto start_indices = rewriter.create<tensor::FromElementsOp>( loc, tensorflow::GetTypeFromTFTensorShape( {static_cast<int64_t>(begin_values.size())}, index_ty), begin_values);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)