- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 155 for created (0.2 sec)
-
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/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/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) -
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/cc/framework/cc_op_gen.cc
TF_CHECK_OK(env->NewWritableFile(dot_h_fname, &h)); TF_CHECK_OK(env->NewWritableFile(dot_cc_fname, &cc)); string op_header_guard; StartFiles(false, dot_h_fname, h.get(), cc.get(), &op_header_guard); // Create the internal versions of these files for the hidden ops. std::unique_ptr<WritableFile> internal_h = nullptr; std::unique_ptr<WritableFile> internal_cc = nullptr; const string internal_dot_h_fname = MakeInternal(dot_h_fname);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 17:22:47 UTC 2023 - 17K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
} result.emplace(std::move(result_content)); return result; } // Used as an argument to TFE_NewCustomDeviceTensorHandle, indicating how // ParallelTensors wrapped in TFE_TensorHandles should be cleaned up once their // reference counts drop to zero. void ParallelTensorDeallocator(void* data) { delete reinterpret_cast<ParallelTensor*>(data); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 29 22:05:31 UTC 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/kernels.cc
class KernelBuilderFactory : public ::tensorflow::kernel_factory::OpKernelFactory { public: explicit KernelBuilderFactory(TF_KernelBuilder* builder) : builder_(builder) {} ::tensorflow::OpKernel* Create( ::tensorflow::OpKernelConstruction* context) override { if (builder_->compute_function) return new ::tensorflow::COpKernel(context, builder_->create_function,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0)