- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 169 for Dtype (0.04 sec)
-
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/tfrt/tests/tf_to_corert/control_flow.mlir
%0 = "tf.Const"() {device = "/device:CPU:0", value = dense<9> : tensor<i32>} : () -> tensor<i32> %1 = "tf.ReadVariableOp"(%arg) {device = "/device:CPU:0", dtype = i32} : (tensor<!tf_type.resource<tensor<i32>>>) -> tensor<i32> %2 = "tf.Less"(%1, %0) {device = "/device:CPU:0"} : (tensor<i32>, tensor<i32>) -> tensor<i1> func.return %2 : tensor<i1> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 00:40:32 UTC 2024 - 17.5K 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/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/parallel_device/parallel_device_lib.h
absl::Span<const int64_t> shape, const TF_DataType dtype) : device_(device), tensors_(std::move(tensors)), shape_(std::vector<int64_t>(shape.begin(), shape.end())), dtype_(dtype) {} ParallelTensor(const ParallelDevice& device, std::vector<TensorHandlePtr> tensors, const TF_DataType dtype) : device_(device), tensors_(std::move(tensors)),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 25 15:21:13 UTC 2023 - 12.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/c/kernels.cc
const char* attr_name, const TF_DataType type, TF_Status* status) { tensorflow::DataType dtype = static_cast<tensorflow::DataType>(type); tensorflow::AddTypeConstraint(kernel_builder, attr_name, dtype, status); } void TF_KernelBuilder_HostMemory(TF_KernelBuilder* kernel_builder,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
const Tensor* value = variable.var()->tensor(); arg.type = value->dtype(); arg.shape = value->shape(); arg.initialized = true; } else { // Instead of embedding constant into HLO, // we handle tf.constant as parameter to reduce size. arg.kind = XlaCompiler::Argument::kParameter; arg.type = input->dtype(); arg.shape = input->shape(); } args.push_back(arg);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K 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/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)