- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for data_type_ (0.31 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
} data_types.push_back(data_type); } n->ClearAttr(type_attr_name); n->AddAttr(type_attr_name, data_types); return data_types; } // Add edges from lifted outside compilation argument nodes to `n` in Graph `g`. void AddEdgesFromOutsideCompilationNodes( const int original_arg_count, const int arg_to_input_edge_offset, const std::vector<DataType>& data_types,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
{{"outside_compilation_O1_host_compute"}, "XlaHostCompute", {"C:o:0", "c:o:0"}, {{"Tinputs", absl::Span<const DataType>({DT_FLOAT, DT_FLOAT})}, {"Toutputs", absl::Span<const DataType>({DT_FLOAT})}, {"ancestors", absl::Span<const string>({})}, {"key", "host_compute_channel_F1_F1_O1"}, {"send_key", ""},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
auto* context = shape_refiner_->GetContext(&node); DataType dtype = node.input_type(idx); return ConvertDataTypeAndShape(dtype, context->input(idx), context->input_handle_shapes_and_types(idx), context, builder); } DataType dtype = node.properties()->input_types[idx]; mlir::Type element_type;
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/c/c_api.cc
desc->node_builder.Attr(attr_name, static_cast<DataType>(value)); } void TF_SetAttrTypeList(TF_OperationDescription* desc, const char* attr_name, const TF_DataType* values, int num_values) { desc->node_builder.Attr( attr_name, ArraySlice<const DataType>( reinterpret_cast<const DataType*>(values), num_values)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
return op.emitOpError("number of segments should be a 0-D tensor"); auto data_type = op.getData().getType().template dyn_cast<RankedTensorType>(); auto segment_ids_type = op.getSegmentIds().getType().template dyn_cast<RankedTensorType>(); if (data_type && segment_ids_type) { if (data_type.getRank() < segment_ids_type.getRank()) return op.emitOpError(
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/lite/ir/tfl_ops.cc
} os << "<unknown TFL type>"; } Type TFLDialect::parseType(DialectAsmParser& parser) const { StringRef data_type; if (parser.parseKeyword(&data_type)) return Type(); if (data_type == "control") return ControlType::get(getContext()); parser.emitError(parser.getNameLoc()) << "unknown TFL type: " << data_type; return nullptr; } void TFLDialect::initialize() { addOperations< #define GET_OP_LIST
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
// trouble calling ConvertToTensor(). For now, extract the tensor data from // ElementsAttr directly in this and read type from tflite::TensorType instead // of tensorflow::DataType. auto type = mlir::cast<TensorType>(value.getType()); tflite::TensorType tflite_element_type = GetTFLiteType(type.getElementType()).value(); if (tflite_element_type == tflite::TensorType_INT4) {
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/tests/tpu_rewrite.mlir
func.func @unsupported_operand_type(%arg0: tensor<?xi2>) { // expected-error@+1 {{failed to determine operand type at index 0: Converting i2 to DataType}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K bytes - Viewed (0)