- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 55 for Dtype (0.09 sec)
-
tensorflow/c/experimental/saved_model/internal/saved_model_api_test.cc
// inputs: { // key : "a" // value: { // name : "serving_default_a:0" // dtype: DT_FLOAT // tensor_shape: { // } // } // } // inputs: { // key : "b" // value: { // name : "serving_default_b:0" // dtype: DT_FLOAT // tensor_shape: { // } // } // } // outputs: {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 08:08:45 UTC 2024 - 21.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc
} tensorflow::DataType dtype; auto status = tensorflow::ConvertToDataType(operand.getType(), &dtype); if (!status.ok()) { op->emitRemark() << "skipping legalization due to " << status.ToString(); return tensorflow::XlaExpression::Invalid(); } return tensorflow::XlaExpression::XlaOp(xla_op, dtype); } } // namespace mhlo
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 18.9K 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/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_distributed_test.cc
" attr {" " key: 'dtype'" " value {" " type: DT_FLOAT" " }" " }" " }" " node_def {" " name: 'read1'" " op: 'ReadVariableOp'" " input: 'var'" " device: '/job:localhost/replica:0/task:1/device:CPU:0'" " attr {" " key: 'dtype'" " value {"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
ctx.appendDialectRegistry(registry); ctx.loadAllAvailableDialects(); } Status ConvertDataTypeToTensor(tensorflow::DataType dtype, Builder builder, Type* type) { Status s = tensorflow::ConvertDataType(dtype, builder, type); if (s.ok()) *type = UnrankedTensorType::get(*type); return s; } class MlirTensor : public TracingTensorHandle { public: explicit MlirTensor(Value value)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor_test.cc
class ConvertTensorTest : public ::testing::Test { protected: template <typename T> void VerifyConversion(std::initializer_list<T> values, DataType dtype, mlir::Type expected_ty) { mlir::Builder b(expected_ty.getContext()); Tensor tensor(dtype, TensorShape({static_cast<int64_t>(values.size())})); tensor.flat<T>().setValues(values); auto value_or = ConvertTensor(tensor, &b);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
} Status ConvertAttribute(mlir::Type type, bool remove_ref_type, AttrValue* value) { DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(type, &dtype)); if (tensorflow::IsRefType(dtype)) dtype = tensorflow::RemoveRefType(dtype); value->set_type(dtype); return absl::OkStatus(); } Status ConvertAttribute(const mlir::TypeAttr& type, bool remove_ref_type,
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.cc
size_t idx = type_and_idx.index(); auto result_ty = mlir::cast<mlir::RankedTensorType>(type_and_idx.value()); // If the result type isn't static, then the owner of the result may be a // cast op from a more specific bounded type to an unbounded dynamic type. // Use the bounded type to get the buffer size. mlir::RankedTensorType buffer_ty = result_ty; if (!buffer_ty.hasStaticShape()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0)