- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 149 for Dtype (0.04 sec)
-
tensorflow/c/experimental/saved_model/core/constant_loading_test.cc
// Get test parameters auto& test_params = GetParam(); DataType dtype = std::get<0>(test_params); TensorShape shape(std::get<1>(test_params)); bool tensorproto_use_tensor_content = std::get<2>(test_params); // Construct a Tensor with the given dtype + shape Tensor expected(dtype, shape); testing::FillNumericTensorBuffer(expected.dtype(), expected.NumElements(), expected.data(), 42);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 4.2K 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/c/checkpoint_reader.cc
Status status; if (reader_ != nullptr) { status = reader_->GetTensor(name, out_tensor); } else { tensorflow::DataType dtype; tensorflow::TensorShape shape; status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape); if (status.ok()) { out_tensor->reset(new Tensor(dtype, shape)); status = v2_reader_->Lookup(name, out_tensor->get()); if (!status.ok()) out_tensor->reset(); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 25 21:29:12 UTC 2023 - 5.5K bytes - Viewed (0) -
tensorflow/c/kernels/bitcast_op_test.cc
def.set_device(DEVICE_CPU); AttrValue typeAttr; SetAttrValue(input_tensor->dtype(), &typeAttr); AttrValue outTypeAttr; SetAttrValue(out_type, &outTypeAttr); (*def.mutable_attr())["T"] = typeAttr; (*def.mutable_attr())["type"] = outTypeAttr; def.add_input( strings::StrCat("input1: ", DataTypeString(input_tensor->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 - 5.5K bytes - Viewed (0) -
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/internal/utils/test_metadata_config.cc
xla_shape.dimensions(), &tensor_shape)); arg_shapes.emplace_back(tensor_shape); DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(input_type, &dtype)); auto metadata_arg = metadata_proto.add_args(); metadata_arg->set_kind(tpu::TPUCompileMetadataProto::Arg::PARAMETER); metadata_arg->set_dtype(dtype); } return absl::OkStatus(); } absl::Status SetupReturnValues(mlir::ModuleOp module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
} std::string ExpectedString(const char* type) { const auto format_str = R"str(kernel { op: "TypeOp%s" device_type: "FakeDeviceName1" constraint { name: "T" allowed_values { list { type: %s } } } } )str"; return absl::StrFormat(format_str, type, type); } #define TEST_KERNEL_TYPE_CONSTRAINT(tf_type, dtype) \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/impl/iostream.cc
std::ostream& operator<<(std::ostream& o, const String& str) { return o << str.str(); } std::ostream& operator<<(std::ostream& o, const TensorSpec& x) { o << "TensorSpec(shape = " << x.shape.DebugString() << ", dtype = " << x.dtype << ")"; return o; } } // namespace impl } // namespace libtf
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 09 21:11:15 UTC 2021 - 1.5K 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/cc/tools/freeze_saved_model.cc
// Identity node that is forwarding the value of a frozen // VarhandleOp. We ensure that the dtype matches of the variable dtype. NodeDef* new_node = frozen_graph_def->add_node(); *new_node = node; (*new_node->mutable_attr())["T"] = name_to_node_map.at(*handle_name)->attr().at("dtype"); continue; } } // If the node isn't a variable, just copy the node as-is.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 11 08:05:36 UTC 2023 - 11.9K bytes - Viewed (0)