- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 149 for Dtype (0.11 sec)
-
tensorflow/compiler/jit/xla_launch_util.cc
// Copy XLA results to the OpOutputList. int output_num = 0; for (int i = 0, end = ctx->num_outputs(); i < end; ++i) { const DataType& type = compilation_result.outputs[i].type; VLOG(2) << "Populating output for retval " << i << " type " << DataTypeString(type); if (type == DT_VARIANT) { return absl::UnimplementedError( "Support for TensorList crossing the XLA/TF boundary "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/device_compiler_test_helper.cc
*graph.add_node() = MakeNode("a", "Placeholder", {}, {{"dtype", DT_FLOAT}, {"shape", input_shape}}); *graph.add_node() = MakeNode("b", "Placeholder", {}, {{"dtype", DT_FLOAT}, {"shape", input_shape}}); *graph.add_node() = MakeNode("c", "Placeholder", {}, {{"dtype", DT_FLOAT}, {"shape", input_shape}});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 08:24:16 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/mangling_util.cc
} Status DemangleTensor(absl::string_view str, TensorProto* proto) { return ParseTextProto(str, kTensorPrefix, proto); } string MangleDataType(const DataType& dtype) { return absl::StrCat(kDataTypePrefix, DataType_Name(dtype)); } Status DemangleDataType(absl::string_view str, DataType* proto) { absl::string_view pbtxt; TF_RETURN_IF_ERROR(ConsumePrefix(str, kDataTypePrefix, &pbtxt));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/test_utils.cc
switch (dtype) { #define CASE(type) \ case DataTypeToEnum<type>::value: { \ type* typed_a = static_cast<type*>(a); \ type* typed_b = static_cast<type*>(b); \ for (int64_t i = 0; i < num_elements; ++i) { \ if (DataTypeIsFloating(dtype)) { \ EXPECT_FLOAT_EQ(static_cast<float>(typed_a[i]), \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 08 20:13:32 UTC 2023 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_tensor_buffer_util.cc
#include "tsl/platform/statusor.h" namespace tensorflow { static size_t GetTensorSize(const TensorShape& shape, const DataType dtype) { return shape.num_elements() * DataTypeSize(dtype); } absl::StatusOr<Tensor> MakeTensorFromPjRtBuffer( const DataType dtype, const TensorShape& shape, std::unique_ptr<xla::PjRtBuffer> pjrt_buffer) { TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::PjRtBuffer::ExternalReference> ref,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 14 18:14:47 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_device_context.cc
shape_determination_fns.layout_preference_fn( cpu_tensor->shape(), cpu_tensor->dtype(), std::nullopt); TF_ASSIGN_OR_RETURN(xla::Shape shape, shape_determination_fns.shape_representation_fn( cpu_tensor->shape(), cpu_tensor->dtype(), /*fast_mem=*/false, layout_preference)); const xla::Layout* device_layout = &(shape.layout());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:49:31 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_metadata_utils.cc
for (auto operand_type_and_idx : llvm::enumerate(op.getOperandTypes())) { Type operand_type = operand_type_and_idx.value(); int index = operand_type_and_idx.index(); tensorflow::tpu::TPUCompileMetadataProto::Arg* arg = metadata->add_args(); tensorflow::DataType dtype; tensorflow::Status status = tensorflow::ConvertToDataType(operand_type, &dtype); if (!status.ok()) return op.emitOpError(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_flags.cc
absl::StrSplit(type, absl::ByAnyChar("()")); // If type has subtypes then parts[0] = type, parts[1] = subtypes, // parts[2] = "" if (parts.size() != 3 && parts.size() != 1) { return errors::InvalidArgument("Invalid type '", type, "'"); } else if (parts.size() == 3) { // First part is the type, second is the subtype ArrayInfo::SubTypeInfo subtype;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/ops/restore_ops.cc
const std::string& checkpoint_key, DataType dtype, ImmediateTensorHandlePtr* out) { // Create the EagerOp ImmediateOpPtr restore_op(ctx->CreateOperation()); TF_RETURN_IF_ERROR(restore_op->Reset("RestoreV2", "/cpu:0")); TF_RETURN_IF_ERROR(restore_op->SetAttrTypeList("dtypes", &dtype, 1)); ImmediateTensorHandlePtr prefix_handle; TF_RETURN_IF_ERROR(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc
return mlir::TF::ShapeAttr::get(type.getContext(), tensor_type.getShape()); } // If type is not a RankedTensor or UnrankedTensor, it must be a scalar. // Empty TensorShape indicates a scalar. return mlir::TF::ShapeAttr::get(type.getContext(), ArrayRef<int64_t>()); } absl::StatusOr<TensorSpecProto> ConvertTypeToTensorSpecProto( const mlir::Type& type) { DataType dtype;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 20.5K bytes - Viewed (0)