- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 149 for Dtype (0.12 sec)
-
tensorflow/cc/framework/ops.cc
} auto const& first = *v.begin(); // Check to make sure that the constituent Initializers are all the same // type and same shape. for (auto const& e : v) { if (e.tensor.dtype() != first.tensor.dtype()) { status = errors::InvalidArgument( "Initializer list components should all have the same type"); return; } if (!TensorShape{e.tensor.shape()}.IsSameSize(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 3.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
} if (TFE_TensorHandleDataType(component.get()) != dtype) { TF_SetStatus(status, TF_INTERNAL, "Components of a ParallelTensor must all have " "the same dtype"); return nullptr; } } return std::unique_ptr<ParallelTensor>( new ParallelTensor(parallel_device, std::move(components), shape, dtype)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 07:47:20 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/c/kernels_experimental.cc
AllocatorAttributes attr; if (cc_input.dtype() == ::tensorflow::DT_VARIANT) { attr.set_on_host(true); } TF_RETURN_IF_ERROR(cc_ctx->allocate_temp(cc_input.dtype(), cc_input.shape(), cc_out, attr)); switch (cc_input.dtype()) { case ::tensorflow::DT_INVALID: { *cc_out = Tensor(::tensorflow::DT_INVALID);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:12:29 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/runtime/runtime.cc
.values(); for (const StructuredValue& arg : args) { PartialTensorShape shape = arg.tensor_spec_value().shape(); DataType dtype = arg.tensor_spec_value().dtype(); TaggedValue tensor_spec(shape, dtype); input_signature.tuple().emplace_back(tensor_spec); } // Setup output signature. TensorSpecProto output_tensor_spec_proto =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 01 11:18:25 UTC 2022 - 7.4K bytes - Viewed (0) -
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/compiler/jit/xla_tpu_device.cc
} se::Stream* const src_compute_stream = src_xla_context->stream(); TF_RET_CHECK(src_compute_stream != nullptr); TF_RET_CHECK(input->dtype() == output->dtype()) << "input type: " << DataTypeString(input->dtype()) << " output type " << DataTypeString(output->dtype()); TF_RET_CHECK(input->shape() == output->shape()); TF_RET_CHECK(DMAHelper::CanUseDMA(input)); auto* const src_compute_stream_impl =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.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/c/eager/c_api_test.cc
name_and_attrs.attr().find("dtype")->second.type()); TF_DeleteBuffer(serialized_attr_values); TFE_Op* var_op_2 = TFE_NewOp(ctx, "VarHandleOp", status); string serialized_dtype; ASSERT_TRUE(name_and_attrs.attr().find("dtype")->second.SerializeToString( &serialized_dtype)); TFE_OpSetAttrValueProto( var_op_2, "dtype",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0)