- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 23 for GetTypeFromTFTensorShape (0.51 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
Value input_shape = rewriter.create<ShapeOp>( loc, tensorflow::GetTypeFromTFTensorShape({4}, rewriter.getI64Type()), input); input_shape = rewriter.create<CastOp>( loc, tensorflow::GetTypeFromTFTensorShape({4}, out_size_element_ty), input_shape); auto scalar_dim_ty = tensorflow::GetTypeFromTFTensorShape({}, out_size_element_ty);
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/tensorflow/transforms/collection_ops_util.cc
auto slice_type = tensorflow::GetTypeFromTFTensorShape( slice_size, buffer_type.getElementType()); auto slice = builder.create<TF::SliceOp>( loc, ArrayRef<Type>{slice_type}, ArrayRef<Value>{buffer, GetIndicesForElement(index, buffer, builder, loc), size_const}); if (keep_slice_shape) return slice; auto element_type = tensorflow::GetTypeFromTFTensorShape(
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/transforms/lower_static_tensor_list.cc
RankedTensorType position_type = tensorflow::GetTypeFromTFTensorShape({-1}, shape_dtype); Value partial_start_position = CreateI32SplatTensor(loc, rewriter, item_rank, 0); Value scalar_zero = CreateI32SplatConst(loc, rewriter, {}, 0); RankedTensorType vector_type = tensorflow::GetTypeFromTFTensorShape({1}, shape_dtype); auto expanded_start_index = rewriter->create<TF::ExpandDimsOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/unfold_large_splat_constant.cc
mlir::arith::ConstantOp fill_shape = op_builder->create<mlir::arith::ConstantOp>( const_op->getLoc(), DenseIntElementsAttr::get( tensorflow::GetTypeFromTFTensorShape( {splat_elements_attr.getType().getRank()}, op_builder->getI64Type()), splat_elements_attr.getType().getShape())); mlir::arith::ConstantOp fill_value =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
OpBuilder::InsertionGuard insertion_point_gurad(*builder); Type input_type = tensorflow::GetTypeFromTFTensorShape(/*shape=*/{}, input_element_type); Type index_type = tensorflow::GetTypeFromTFTensorShape(/*shape=*/{}, index_element_type); Block *block = builder->createBlock(body); Location loc = body->getLoc();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dynamic_shape_utils.cc
llvm::ArrayRef<int64_t> shapes) { return llvm::to_vector(llvm::map_range(shapes, [](int64_t shape) { return mlir::ShapedType::isDynamic(shape) ? kTFDynamicSize : shape; })); } mlir::RankedTensorType GetTypeFromTFTensorShape(llvm::ArrayRef<int64_t> shape, mlir::Type elementType, mlir::Attribute encoding) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 21 16:21:18 UTC 2022 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/const_tensor_utils.cc
tensor.shape_signature.end()); return tensorflow::GetTypeFromTFTensorShape(shape, elem_type); } if (!tensor.shape.empty()) { llvm::SmallVector<int64_t, 4> shape(tensor.shape.begin(), tensor.shape.end()); return tensorflow::GetTypeFromTFTensorShape(shape, elem_type); } return UnrankedTensorType::get(elem_type); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 16.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
auto elem_type = filter_type.getElementType(); auto result_type = tensorflow::GetTypeFromTFTensorShape(result_shape, elem_type); // TensorFlow Lite `Reshape` op only support int32 shape tensor currently. auto shape_type = tensorflow::GetTypeFromTFTensorShape({4}, rewriter.getIntegerType(32)); SmallVector<Attribute, 4> result_shape_data(4); for (int i = 0; i < 4; ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dynamic_shape_utils.h
llvm::SmallVector<int64_t> ConvertTFShapeToMlir(llvm::ArrayRef<int64_t> shapes); llvm::SmallVector<int64_t> ConvertMlirShapeToTF(llvm::ArrayRef<int64_t> shape); static constexpr int64_t kTFDynamicSize = -1; mlir::RankedTensorType GetTypeFromTFTensorShape(llvm::ArrayRef<int64_t> shape, mlir::Type elementType, mlir::Attribute encoding = {}); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 18 11:40:17 UTC 2022 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
return RangeOp::build( builder, result, tensorflow::GetTypeFromTFTensorShape( size.getSExtValue(), start.getType().cast<TensorType>().getElementType()), start, limit, delta); } return RangeOp::build( builder, result, tensorflow::GetTypeFromTFTensorShape( {-1}, start.getType().cast<TensorType>().getElementType()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)