Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 121 for vecotr (0.14 sec)

  1. tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen_main.cc

    #include "tensorflow/core/platform/types.h"
    #include "tsl/platform/status.h"
    
    namespace tensorflow {
    namespace cc_op {
    namespace {
    
    void WriteAllFuzzers(string root_location, std::vector<string> api_def_dirs,
                         std::vector<string> op_names) {
      OpList ops;
      absl::StatusOr<ApiDefMap> api_def_map =
          LoadOpsAndApiDefs(ops, false, api_def_dirs);
    
      TF_CHECK_OK(api_def_map.status());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 10:53:49 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/kernels/xla_ops.cc

    std::vector<int> ConstantsVector(OpKernelConstruction* ctx) {
      DataTypeVector constant_types;
      OP_REQUIRES_OK_RETURN(ctx, std::vector<int>(),
                            ctx->GetAttr("Tconstants", &constant_types));
      std::vector<int> constants(constant_types.size());
      std::iota(constants.begin(), constants.end(), 0);
      return constants;
    }
    
    std::vector<int> ResourcesVector(OpKernelConstruction* ctx) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 22:46:36 UTC 2024
    - 41.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tf_graph_optimization_pass.cc

                           module_out->getBody()->getOperations());
    }
    
    // Returns a vector of passes from their names. If a pass is not found, then the
    // corresponding return entry is null.
    static std::vector<GraphOptimizationPass*> FindRegisteredPassesByName(
        const std::vector<std::string>& pass_names) {
      std::vector<GraphOptimizationPass*> pass_ids(pass_names.size(), nullptr);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 7.6K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/ops/gen/cpp/views/op_view.cc

          OpArgumentView("const char*", "raw_device_name", "nullptr"));
    }
    
    const std::vector<ArgView>& OpView::Inputs() const { return input_args_; }
    
    const std::vector<ArgView>& OpView::Outputs() const { return output_args_; }
    
    const std::vector<AttrView>& OpView::Attributes() const {
      return argument_attrs_;
    }
    
    const std::vector<OpArgumentView>& OpView::AllArguments() const {
      return all_arguments_;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 03 07:02:00 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc

          RankedTensorType::get(ret.size(), builder.getIntegerType(32)), ret);
    }
    
    bool DenseI64AttrToI32Vector(const DenseIntElementsAttr& dense_attr,
                                 std::vector<int32_t>* out_vec) {
      std::vector<int32_t> ret(dense_attr.getNumElements());
      auto range = dense_attr.getValues<int64_t>();
      std::transform(range.begin(), range.end(), ret.begin(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 18:33:05 UTC 2024
    - 3.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.cc

        const XlaShapeLayoutHelpers::ShapeDeterminationFns
            shape_determination_funcs,
        const std::vector<tensorflow::TensorShape>& arg_shapes,
        std::vector<tpu::ShardingAndIndex>* arg_core_mapping,
        std::vector<std::vector<xla::Shape>>* per_core_arg_shapes,
        xla::CompileOnlyClient* client,
        XlaCompiler::CompilationResult* compilation_result) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 14K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc

      auto client =
          xla::ClientLibrary::GetOrCreateCompileOnlyClient(cpu_platform).value();
    
      std::vector<TensorShape> arg_shapes;
      TPUCompileMetadataProto metadata_proto;
      bool use_tuple_args = true;
      std::vector<ShardingAndIndex> arg_core_mapping;
      std::vector<std::vector<xla::Shape>> per_core_arg_shapes;
      std::vector<std::unique_ptr<mlir::Pass>> custom_legalization_passes;
    
      // This doesn't actually compile correctly.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 13 23:59:33 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/python/saved_model_to_tfl_flatbuffer.cc

      mlir::MLIRContext context;
      mlir::quant::QuantizationSpecs quant_specs;
    
      // Parse input arrays.
      std::vector<string> node_names;
      std::vector<string> node_dtypes;
      std::vector<std::optional<std::vector<int>>> node_shapes;
      std::vector<std::optional<double>> node_mins;
      std::vector<std::optional<double>> node_maxs;
    
      // Populate quantization specs.
      TF_RETURN_IF_ERROR(internal::PopulateQuantizationSpecs(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun May 12 12:39:37 UTC 2024
    - 11K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.h

        *out_attr = content;
        return true;
      } else {
        return false;
      }
    }
    
    // Changes a DenseIntElementsAttr **containing I64** elements to an I32 Vector.
    bool DenseI64AttrToI32Vector(const DenseIntElementsAttr& dense_attr,
                                 std::vector<int32_t>* out_vec);
    
    // Gets boolean from composite attrs if it exists.
    std::optional<bool> GetBoolFromCompositeAttr(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 18:33:05 UTC 2024
    - 3.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter_test.cc

                           fb_builder.CreateVector(std::vector<float>({1.0, 5.0}))),
          CreateOpMetadata(fb_builder, 2, 0,
                           fb_builder.CreateVector(std::vector<float>({1.0, 5.0}))),
          CreateOpMetadata(
              fb_builder, 3, 1,
              fb_builder.CreateVector(std::vector<float>({-1.0, 2.0}))),
      };
      const auto subgraphs = {CreateSubgraphMetadata(
          fb_builder, fb_builder.CreateVector(ops.begin(), ops.size()))};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 06:11:34 UTC 2024
    - 6K bytes
    - Viewed (0)
Back to top