- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 149 for Dtype (0.04 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
VLOG(1) << "[potentially conservative] Op type `" << node.type_string() << "` is stateful but effects not modelled"; } else { // See if any resource type is used. bool resource = false; std::function<bool(mlir::Type)> record_resource; record_resource = [&](mlir::Type type) { type.walk([&](mlir::Type t) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K 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/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) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
operation->SetOpName(absl::StrCat("_input_", inputs_.size()).c_str())); TF_RETURN_IF_ERROR(operation->SetAttrType("dtype", dtype)); if (!shape.unknown_rank()) { TF_RETURN_IF_ERROR(operation->SetAttrShape( "shape", reinterpret_cast<int64_t*>(shape.dim_sizes().data()), shape.dims())); } int num_outputs = 1; std::vector<AbstractTensorHandle*> outputs(num_outputs);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 20:00:09 UTC 2024 - 15.4K 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) -
tensorflow/compiler/aot/aot_only_var_handle_op.cc
.Attr("shared_name: string = ''") .Attr("debug_name: string = ''") .Attr("dtype: type") .Attr("shape: shape") .Output("resource: resource") .SetIsStateful() .SetShapeFn([](shape_inference::InferenceContext* c) { c->set_output(0, c->Scalar()); DataType t; TF_RETURN_IF_ERROR(c->GetAttr("dtype", &t)); PartialTensorShape p; TF_RETURN_IF_ERROR(c->GetAttr("shape", &p));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 09:57:04 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
dtype == tensorflow::DT_UINT16 || dtype == tensorflow::DT_INT16 || dtype == tensorflow::DT_UINT8 || dtype == tensorflow::DT_INT8 || dtype == tensorflow::DT_HALF || dtype == tensorflow::DT_BFLOAT16 || dtype == tensorflow::DT_FLOAT || dtype == tensorflow::DT_DOUBLE || dtype == tensorflow::DT_COMPLEX64 || dtype == tensorflow::DT_COMPLEX128 || dtype == tensorflow::DT_BOOL) { return {};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
for (auto type : inst.getResultTypes()) if (!mlir::isa<mlir::TensorType, mlir::tf_executor::ControlType, mlir::tf_executor::TokenType>(type)) return errors::InvalidArgument( "Values must be of tensor type, TensorFlow control type, or " "TensorFlow token type. Found ", mlir::debugString(type));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
for (auto type : inst.getResultTypes()) if (!mlir::isa<mlir::TensorType, mlir::tf_executor::ControlType, mlir::tf_executor::TokenType>(type)) return errors::InvalidArgument( "Values must be of tensor type, TensorFlow control type, or " "TensorFlow token type. Found ", mlir::debugString(type));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0)