- Sort Score
- Result 10 results
- Languages All
Results 11 - 13 of 13 for argTypes (0.32 sec)
-
tensorflow/compiler/jit/xla_kernel_creator.cc
// Create the kernel. Device* dev = flr->device(); Status s; auto props = std::make_shared<NodeProperties>( &fbody->record->fdef().signature(), node_def, fbody->arg_types, fbody->ret_types); OpKernelConstruction construction(DeviceType(dev->device_type()), dev, dev->GetAllocator(AllocatorAttributes()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 22:24:01 UTC 2023 - 4.4K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen_util.h
string GetOpAttrStruct() const; string GetConstructorDecl(StringPiece op_name_prefix, bool include_attr) const; string op_name; std::vector<string> arg_types; std::vector<string> arg_names; std::vector<string> output_types; std::vector<string> output_names; std::vector<bool> is_list_output; bool has_optional_attrs; string comment;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 00:57:05 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_restore_op.cc
kTfSavedModelIndexPathAttr, builder.getArrayAttr({file_prefix_attr}))}); const int insert_idx = func_op.getNumArguments(); func_op.insertArgument(insert_idx, /*argType=*/filename_op_type, arg_attrs, NameLoc::get(file_prefix_attr)); return func_op.getArgument(insert_idx); } // Creates a 1D string array constant for "tensor_names" input of `RestoreV2`
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Mar 12 06:02:20 UTC 2023 - 9K bytes - Viewed (0)