- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for Slice_types (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
auto slice_shape = llvm::to_vector<8>(buffer_type.getShape()); slice_shape[0] = 1; auto slice_type = tensorflow::GetTypeFromTFTensorShape( slice_shape, buffer_type.getElementType()); auto update_slice = element; if (element.getType() != slice_type) { update_slice = builder.create<TF::ReshapeOp>( loc, ArrayRef<Type>{slice_type}, ArrayRef<Value>{element, GetR1Const(slice_shape, builder, loc)}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
slice_size.push_back(-1); } } auto slice_type = mlir::RankedTensorType::get(slice_begin.size(), rewriter->getIntegerType(32)); auto slice_begin_attr = mlir::DenseIntElementsAttr::get(slice_type, slice_begin); auto slice_size_attr = mlir::DenseIntElementsAttr::get(slice_type, slice_size); auto slice_begin_const =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0)