- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 141 for created (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
for (const auto& device : devices) if (DeviceNameUtils::IsCompleteSpecification(spec, device)) matching_devices.push_back(device); return matching_devices; } // Create error message for a conflicting attribute of a device. template <typename T> absl::Status MismatchedTPUSystemAttributeErr(absl::string_view attribute, T a, T b) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_metadata_utils.cc
"bad '{0}' attribute at index {1} with value '{2}': failed to parse to {3}"; constexpr char kBadArrayAttrLengthMsg[] = "bad '{0}' attribute, expected array attribute of size {1}, got size {2}"; // Creates a missing attribute error message. std::string CreateMissingAttributeMsg(llvm::StringRef attribute) { return llvm::formatv("requires attribute '{0}'", attribute).str(); }
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/tensorflow/transforms/decompose_reduce_dataset.cc
} auto dataset_while = builder.create<TF::WhileRegionOp>( reduce_dataset.getLoc(), while_input_types, /*input=*/while_input_values, /*parallel_iterations=*/10, false, /*shape_invariant=*/false); // `_lower_using_switch_merge` is the default for While ops created // in TensorFlow and allows lowering to V1 control flow for loop // parallelization.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc
// can't lower tf.Const. Value cst; if (use_mhlo_const) { cst = builder.create<mhlo::ConstantOp>(func_op->getLoc(), attrs); } else { cst = builder.create<TF::ConstOp>(func_op->getLoc(), attrs); } func_op.getArgument(i).replaceAllUsesWith(cst); } return module_op; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 01:03:21 UTC 2024 - 35.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
projection_type, layer_norm_scale_type}; auto func_type = builder->getFunctionType(input_types, output_type); auto func = func::FuncOp::create( mlir::NameLoc::get(builder->getStringAttr("fused_func")), "fused_func", func_type, {}); func.addEntryBlock(); std::vector<std::string> attributes; if (ln) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc
for (int i = 0; i < custom_op.getNumResults(); i++) { output_tys.push_back(custom_op.getType(i)); } op_state.addTypes(output_tys); op_state.addAttributes(attr); auto stablehlo_op = builder.create(op_state); custom_op.replaceAllUsesWith(stablehlo_op); custom_op.erase(); }); } std::unique_ptr<OperationPass<func::FuncOp>> CreateTflToStablehloPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
for (auto attr : op.getPlatforms().getAsRange<StringAttr>()) { platforms.push_back(attr.getValue().str()); } TF_ASSIGN_OR_RETURN( auto loader, tensorflow::XlaCallModuleLoader::Create( context, static_cast<int>(op.getVersion()), op.getModule().str(), std::move(disabled_checks), std::move(platforms), /*num_invocation_args=*/op.getArgs().size(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.cc
// checkpoint when fetched. saver_def.set_save_tensor_name(absl::StrCat(save_node_name, ":0")); return saver_def; } else { return absl::InternalError( absl::StrCat("Failed to create SaverDef. Fields should be either all " "empty strings or all non-empty strings. Got fields: ", absl::StrJoin(fields, ","))); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/c/tf_tensor.cc
// -------------------------------------------------------------------------- static void DeleteArray(void* data, size_t size, void* arg) { DCHECK_EQ(data, arg); delete[] reinterpret_cast<char*>(arg); } // Create an empty tensor of type 'dtype'. 'shape' can be arbitrary, but has to // result in a zero-sized tensor. static TF_Tensor* EmptyTensor(TF_DataType dtype, const tensorflow::TensorShape& shape) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 21:57:32 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
std::vector<Output>* grad_outputs) { // Propagate the incoming gradient along all the selected values, // and zero everywhere else. Use the Pad operator for this. // // First create an Nx2 padding where N is the number of input // dimensions. The first column is the number of prepended zeros // for each dimension, and the second column is the number of // appended zeros. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0)