- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for add_shape (0.84 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
list->add_s(nested_value.s()); break; case AttrValue::kType: list->add_type(nested_value.type()); break; case AttrValue::kShape: *list->add_shape() = nested_value.shape(); break; default: return errors::Unimplemented("Unhandled nested attribute!"); } } else if (auto attr = mlir::dyn_cast<mlir::ElementsAttr>(a)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
AttrValueMap* values) { AttrValue value; auto& shape_list = *value.mutable_list(); for (const std::optional<llvm::ArrayRef<int64_t>>& shape : shapes) { TensorShapeProto& tshape = *shape_list.add_shape(); if (shape.has_value()) { for (int64_t dim : *shape) { tshape.add_dim()->set_size(mlir::ShapedType::isDynamic(dim) ? -1 : dim); } } else { tshape.set_unknown_rank(true); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
handle_dtypes_attr.mutable_list()->add_type(dtype); SetTensorShapeProto(subtype, handle_shapes_attr.mutable_list()->add_shape()); } (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr; (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr; } } TF_RETURN_IF_ERROR(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
handle_dtypes_attr.mutable_list()->add_type(dtype); SetTensorShapeProto(subtype, handle_shapes_attr.mutable_list()->add_shape()); } (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr; (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr; } } TF_RETURN_IF_ERROR(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference.cc
const InferredShape& arg_shape = it->second; shape_inference::InferenceContext* context = shape_refiner->GetContext(n); if (arg_shape.handle_type != DT_INVALID) { shape_inference::ShapeHandle handle; TF_RETURN_IF_ERROR(context->MakeShapeFromPartialTensorShape( arg_shape.handle_shape, &handle));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
absl::StatusOr<TensorShape> GetTensorShapeFromXlaArgument( const XlaArgument& arg) { if (absl::holds_alternative<xla::Shape>(arg.shape)) { TensorShape arg_shape; TF_RETURN_IF_ERROR( XLAShapeToTensorShape(std::get<xla::Shape>(arg.shape), &arg_shape)); return arg_shape; } else { return std::get<TensorShape>(arg.shape); } } Status MaybeRewriteLayoutWithShardedShape( mlir::StringAttr sharding,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
auto paddings_value = builder->create<TF::ConstOp>(filter.getLoc(), paddings); std::vector<int64_t> pad_shape = {filter_shape[0] + pad_h, filter_shape[1] + pad_w, filter_shape[2], filter_shape[3]}; SmallVector<int64_t, 4> expand_shape(pad_shape.begin(), pad_shape.end()); auto expand_result_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0)