Search Options

Results per page
Sort
Preferred Languages
Advance

Results 71 - 80 of 299 for pushBack (0.15 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc

      // `replicate`.
      llvm::SmallVector<Value, 8> reformat_operands;
      for (const auto& entry : execute_arg_to_outer_args) {
        reformat_operands.push_back(execute.getArgs()[entry.first]);
      }
      reformat_operands.push_back(compile_launch.getResult(1));
      reformat_operands.push_back(replicate.GetBody().getArgument(
          replicate.GetNumReplicatedBlockArguments() - 1));
      builder.setInsertionPoint(execute_launch);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/call_graph_util.cc

        // each entry function. The one exception are initializer functions, which
        // have `tf_saved_model.initializer_type` instead.
        if (IsEntryFunction(func)) {
          entry_funcs.push_back(func);
        }
      });
      return entry_funcs;
    }
    
    LogicalResult GetCallees(SymbolUserOpInterface op, SymbolTable &symtab,
                             llvm::SmallVector<func::FuncOp> &callees) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 2.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/common/ir/FakeQuantSupport.cc

        if (std::fabs(rmax - rmin) < std::numeric_limits<double>::epsilon()) {
          scales.push_back(1.0);
          zeroPoints.push_back(qmin);
          continue;
        }
    
        double scale;
        int64_t nudgedZeroPoint;
        getNudgedScaleAndZeroPoint(qmin, qmax, rmin, rmax, scale, nudgedZeroPoint);
        scales.push_back(scale);
        zeroPoints.push_back(nudgedZeroPoint);
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 11:52:27 UTC 2024
    - 7.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc

        SmallVector<int64_t, 4> non_unit_dims;
    
        for (int i = 0; i < input_shape_extended.size(); i++) {
          int64_t dim = broadcast_shape[i];
          if (dim != 1) {
            non_unit_dims.push_back(dim);
            input_carryover_dims.push_back(input_shape_extended[i]);
          }
        }
    
        // If the reshape rank is less than the number of non-unit dimensions
        // of the broadcast, then the reshape collapses non-unit dimensions.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_function.cc

        for (int i = 0; i < noutputs; ++i) {
          output_names_vec.push_back(string(output_names[i]));
        }
      }
    
      // Process control output names.
      std::vector<string> control_output_names_vec;
      if (control_output_names) {
        control_output_names_vec.reserve(ncontrol_outputs);
        for (int i = 0; i < ncontrol_outputs; ++i) {
          control_output_names_vec.push_back(string(control_output_names[i]));
        }
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 13.6K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/ops/gen/cpp/renderers/op_renderer.cc

        if (context_.mode == RendererContext::kHeader) {
          absl::StrAppend(&text, argument.Initializer());
        }
        if (argument.HasDefaultValue()) {
          args_with_default_val.push_back(text);
        } else {
          args_without_default_val.push_back(text);
        }
      }
      std::vector<string> arguments;
      arguments.reserve(args_without_default_val.size() +
                        args_with_default_val.size());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 28 05:51:40 UTC 2024
    - 2.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/deadness_analysis.cc

          absl::string_view frame_name = control_flow_info_[out_id].frame_name;
          if (IsRootEnter(out)) {
            ready_enters_per_frame[frame_name].push_back(out);
          } else if (IsRootExit(out)) {
            ready_exits.push_back(out);
          } else {
            ready.push_back(out);
          }
        }
    
        if (ready.empty()) {
          // Try moving nodes from ready_enters_per_frame and ready_exits to
          // `ready`.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 60.4K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc

          if (is_guaranteed_constant(*n)) {
            return errors::InvalidArgument(
                "Guaranteed constants are not supported (", n->name(), ")");
          }
          args.push_back(n);
        } else if (n->type_string() == "_Retval") {
          retvals.push_back(n);
        }
      }
    
      if (std::find(args.begin(), args.end(), nullptr) != args.end()) {
        return errors::InvalidArgument("Missing or non-consecutive arguments");
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc

            for (int i = 0; i < repeats; ++i)
              state_->types.push_back(UnrankedTensorType::get(type_attr.getType()));
          } else if (output_arg.type() != tensorflow::DT_INVALID) {
            for (int i = 0; i < repeats; ++i) {
              Type type;
              TF_RETURN_IF_ERROR(
                  ConvertDataType(output_arg.type(), builder, &type));
              state_->types.push_back(type);
            }
          } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/sparsity/sparsify_model_test.cc

          std::vector<uint8_t>(expected_value.begin(), expected_value.end());
      input_model.buffers.push_back(std::move(model_metadata_buffer));
      auto metadata_t = std::make_unique<tflite::MetadataT>();
      metadata_t->name = tflite::optimize::kTfLiteReducedPrecisionKey;
      metadata_t->buffer = input_model.buffers.size() - 1;
      input_model.metadata.push_back(std::move(metadata_t));
    
      // Sparsify and create output model
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 10 20:16:40 UTC 2024
    - 2.9K bytes
    - Viewed (0)
Back to top