- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for add_shape (0.16 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/tf2xla/api/v1/compile_mlir_util_test.cc
.Attr("dtype", data_type) .Attr("value", tensor); // Create a bad output shape attr. AttrValue shape_attr; TensorShapeProto* shape_proto = shape_attr.mutable_list()->add_shape(); shape_proto->add_dim()->set_size(1); builder.Attr("_output_shapes", shape_attr); TF_RETURN_IF_ERROR(builder.Finalize(&node)); return CreateSingleOpGraph(node, {}, {DataType::DT_INT32}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 19:54:38 UTC 2024 - 9.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/lite/schema/schema_generated.h
}; struct VariantSubTypeBuilder { typedef VariantSubType Table; ::flatbuffers::FlatBufferBuilder &fbb_; ::flatbuffers::uoffset_t start_; void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape) { fbb_.AddOffset(VariantSubType::VT_SHAPE, shape); } void add_type(tflite::TensorType type) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M 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/set_tpu_infeed_layout.cc
// TODO(kramm): Move this into a separate pass. See b/184944903 xla::Shape old_shape = xla::TypeToShape(t); XLA_Shape old_shape_c = {}; XLA_Shape new_shape_c = {}; TfTpu_ExecutorApiFn *executor = stream_executor::tpu::ExecutorApiFn(); if (!stream_executor::tpu::IsInitialized(executor)) { return failure(); } ApiConverter::ToC(old_shape, &old_shape_c);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/legalize_tf_quant_test.cc
auto client = xla::ClientLibrary::GetOrCreateCompileOnlyClient(platform).value(); tensorflow::tpu::TPUCompileMetadataProto metadata_proto; // Set up an arg per arg_shape with the specified type. for (int i = 0; i < arg_shapes.size(); ++i) { auto metadata_arg = metadata_proto.add_args(); metadata_arg->set_kind( tensorflow::tpu::TPUCompileMetadataProto::Arg::PARAMETER);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 18:43:55 UTC 2024 - 7.2K bytes - Viewed (0)