- Sort Score
- Result 10 results
- Languages All
Results 111 - 120 of 263 for created (0.08 sec)
-
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/framework/cc_op_gen_util.cc
is_list_output.push_back(is_list); } strings::StrAppend(&comment, "\nReturns:\n"); if (graph_op_def.output_arg_size() == 0) { // No outputs. strings::StrAppend(&comment, "* the created `Operation`\n"); } else if (graph_op_def.output_arg_size() == 1) { // One output if (is_list_output[0]) { strings::StrAppend(&comment, "* `OutputList`: "); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 00:57:05 UTC 2024 - 25K 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) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc
EXPECT_EQ(compilation_status.Delta(kOldBridgeWithFallbackModeSuccess), 0); } TEST(LegalizeTFTest, RecordsStreamzForNoMlirFallback) { FunctionDef my_func = tensorflow::FunctionDefHelper::Create("empty", {}, {}, {}, {}, {}); tensorflow::FunctionDefLibrary fdef; *(fdef.add_function()) = my_func; tensorflow::FunctionLibraryDefinition flib_def( tensorflow::OpRegistry::Global(), fdef);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/c/c_api_function_test.cc
void RunT(const std::vector<std::pair<TF_Operation*, TF_Tensor*>>& inputs, std::initializer_list<TF_Output> outputs, const std::vector<std::vector<int32_t>>& expected_results) { // Create a session for this graph CSession csession(host_graph_, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); // Run csession.SetInputs(inputs); csession.SetOutputs(outputs);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 20 22:08:54 UTC 2023 - 63.6K bytes - Viewed (0)