- Sort Score
- Result 10 results
- Languages All
Results 11 - 18 of 18 for GetSubtypes (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
// Get the data type of the resource. auto subtypes = mlir::cast<TF::ResourceType>( getElementTypeOrSelf(execute.getOperand(it->second))) .getSubtypes(); if (subtypes.size() != 1) continue; auto data_type = getElementTypeOrSelf(subtypes[0]); // The XLA backend does not yet support formatting 64-bit data types.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
if (mlir::isa<mlir::TF::StringType>(type)) return true; auto sub_type = mlir::dyn_cast<mlir::TF::TensorFlowTypeWithSubtype>(type); if (!sub_type) return false; bool has_string = llvm::any_of(sub_type.GetSubtypes(), [](TensorType type) { return mlir::isa<mlir::TF::StringType>(type.getElementType()); }); return has_string; } bool HasStringOperand(Operation& op) { for (auto operand : op.getOperands()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
if (auto resource_type = mlir::dyn_cast<mlir::TF::ResourceType>(arg_type.getElementType())) { llvm::ArrayRef<mlir::TensorType> subtypes = resource_type.getSubtypes(); if (!subtypes.empty()) { AttrValue handle_dtypes_attr; AttrValue handle_shapes_attr; for (mlir::TensorType subtype : subtypes) { DataType dtype;
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
if (auto resource_type = mlir::dyn_cast<mlir::TF::ResourceType>(arg_type.getElementType())) { llvm::ArrayRef<mlir::TensorType> subtypes = resource_type.getSubtypes(); if (!subtypes.empty()) { AttrValue handle_dtypes_attr; AttrValue handle_shapes_attr; for (mlir::TensorType subtype : subtypes) { DataType dtype;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
"return getElementTypeOrSelf(element_type());">; let extraClassDeclaration = [{ // Returns type of the TensorList element produced by this op. TensorType element_type() { return handle_dtype().getSubtypes()[0]; } // Returns data type of the result handle. Returned type contains type of // the TensorList element as a subtype. VariantType handle_dtype() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
} // We only support up to one nested type in tf_type.variant_type. if (variant_type.getSubtypes().size() > 1) { return std::nullopt; } if (variant_type.getSubtypes().empty()) { return variant_params; } mlir::TensorType tensor_type = variant_type.getSubtypes().front(); tflite::TensorType tflite_element_type = GetTFLiteType(tensor_type.getElementType()).value();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
// no single type in the resource. Type GetResourceSubtype(Value value) { auto resource_type = mlir::dyn_cast<TF::ResourceType>(getElementTypeOrSelf(value.getType())); auto subtypes = resource_type.getSubtypes(); if (subtypes.size() == 1) return subtypes[0]; return nullptr; } // Replaces all `tf.VarIsInitializedOp` in a block with a constant true. // TODO(b/171039585): Replace this with proper analysis of
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// This is required to populate derived attributes during export in a // meaningful way. Else during export to GraphDef element_type() query // will result in out of bounds access/assert. if (handle_dtype().getSubtypes().size() != 1) { return emitOpError( "must have exactly one subtype in the result variant type"); } if (!IsOfRankOrUnranked(op.getElementShape(), 0) &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0)