Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 13 of 13 for argTypes (0.32 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top