- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 54 for Dtype (0.04 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/convert_type_test.cc
bool unknown_rank, DataType dtype) { TensorShapeProto shape; shape.set_unknown_rank(unknown_rank); for (int64_t dim : dims) { shape.add_dim()->set_size(dim); } mlir::MLIRContext context; mlir::Builder b(&context); auto status_or = ConvertToMlirTensorType(shape, dtype, &b); std::string buf; llvm::raw_string_ostream os(buf); status_or.value().print(os);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_config.cc
#include "llvm/Support/Debug.h" #include "llvm/Support/raw_ostream.h" #include "tensorflow/core/framework/types.pb.h" // Returns whether the given dtype is a quantization type in TensorFlow. static bool IsQuantizationType(tensorflow::DataType dtype) { switch (dtype) { case tensorflow::DT_QINT8: case tensorflow::DT_QUINT8: case tensorflow::DT_QINT16: case tensorflow::DT_QUINT16: case tensorflow::DT_QINT32:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
TF_DeleteExecutionContext(ctx); return wrap(func); } TF_AbstractTensor* TF_AddFunctionParameter(TF_ExecutionContext* func, TF_DataType dtype, TF_Shape shape, TF_Status* s) { DCHECK_GE(shape.num_dims, -1); TracingTensorHandle* t; TracingContext* tracing_ctx = dyn_cast<TracingContext>(unwrap(func));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 10:15:17 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration_test.cc
func.func @main() -> (tensor<1x2xf16>, tensor<2xf16>) { %graph:2 = tf_executor.graph { %0:2 = tf_executor.island wraps "tf.Const"() {device = "", dtype = "tfdtype$DT_HALF", value = dense<1.0> : tensor<1x2xf16>} : () -> tensor<1x2xf16> loc("const1") %1:2 = tf_executor.island wraps "tf.Const"() {device = "", dtype = "tfdtype$DT_HALF", value = dense<[1.0, 2.0]> : tensor<2xf16>} : () -> tensor<2xf16> loc("const2")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 20 23:11:32 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/saved_model/saved_model.cc
absl::StatusOr<std::pair<tensorflow::DataType, tensorflow::PartialTensorShape>> ProcessTensorSpec(mlir::TensorType type) { tensorflow::DataType dtype; TF_RETURN_IF_ERROR( ConvertScalarTypeToDataType(type.getElementType(), &dtype)); if (!type.hasRank()) return std::make_pair(dtype, tensorflow::PartialTensorShape()); auto shape = type.getShape(); llvm::SmallVector<int64_t, 4> dims; dims.assign(shape.begin(), shape.end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_recv_device_context.cc
Tensor* cpu_tensor, StatusCallback done) { DataType dtype = EncodePrimitiveTypeAsDataType(shape_.element_type()).value(); TensorShape tensor_shape; Status status = XLAShapeToTensorShape(shape_, &tensor_shape); if (!status.ok()) { done(status); return; } *cpu_tensor = Tensor(dtype, tensor_shape); status = stream_->Memcpy(cpu_tensor->data(), device_memory_base_,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 1.9K bytes - Viewed (0) -
tensorflow/c/kernels/summary_op_test.cc
def.set_device(DEVICE_CPU); AttrValue valuesTypeAttr; SetAttrValue(values->dtype(), &valuesTypeAttr); (*def.mutable_attr())["T"] = valuesTypeAttr; def.add_input(strings::StrCat("input1: ", DataTypeString(tags->dtype()))); def.add_input(strings::StrCat("input2: ", DataTypeString(values->dtype()))); std::unique_ptr<OpKernel> kernel =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 6.7K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
tensorflow::AsyncValueTensor* av_tensor = tensorflow::AsyncValueTensor::FromTensor(tensor); if (av_tensor == nullptr) { TF_ASSIGN_OR_RETURN( *tensor, MakeTensorFromPjRtBuffer(tensor->dtype(), tensor->shape(), std::move(buffer))); } else { av_tensor->SetBuffer(std::move(buffer)); } return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 05:48:24 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/cc/ops/while_loop.cc
NodeBuilder::NodeOut enter_input(enter_output.node(), enter_output.index()); const int next_output_index = 0; DataType dtype = enter_output.node()->output_type(0); NodeBuilder::NodeOut next_input(NextIterationName(scope, loop_var_idx), next_output_index, dtype); std::vector<NodeBuilder::NodeOut> input_list({enter_input, next_input}); const string unique_name = scope.GetUniqueNameForOp("Merge");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 01:01:21 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dump_graph_test.cc
} TEST(Dump, TextualIrWithOptions) { Graph graph(OpRegistry::Global()); Node* node; TF_ASSERT_OK(NodeBuilder("A", "Placeholder") .Attr("dtype", DT_FLOAT) .Finalize(&graph, &node)); string actual; StringWritableFile file(&actual); TF_ASSERT_OK(DumpTextualIRToFile(MlirDumpConfig().emit_location_information(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 3.6K bytes - Viewed (0)