- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for begin_values (0.25 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)