Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 192 for pushBack (0.19 sec)

  1. tensorflow/cc/experimental/libtf/tests/value_test.cc

      TaggedValue valuef(3.f);
      TaggedValue valuei(int64_t(3));
      TaggedValue list = TaggedValue::List();
      TaggedValue tuple = TaggedValue::Tuple();
      tuple.tuple().push_back(TaggedValue(int64_t(310)));
      list.list().push_back(valuei);
      list.list().push_back(valuef);
      list.list().push_back(tuple);
      std::stringstream stream;
      stream << list;
      ASSERT_EQ(stream.str(), "[3, 3, (310, ), ]");
    }
    
    TEST(ValueTest, TestString) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Sep 01 11:18:25 UTC 2022
    - 3.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/representative_dataset_test.cc

      *(config_1.mutable_tf_record()->mutable_path()) = "test_path_1";
      representative_dataset_configs.push_back(config_1);
    
      // Signature key is implicitly "serving_default".
      RepresentativeDatasetConfig config_2{};
      *(config_2.mutable_tf_record()->mutable_path()) = "test_path_2";
      representative_dataset_configs.push_back(config_2);
    
      const absl::StatusOr<
          absl::flat_hash_map<std::string, RepresentativeDatasetFile>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jan 29 04:53:21 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  3. tensorflow/cc/gradients/manip_grad.cc

      auto shift = op.input(1);
      auto axis = op.input(2);
      auto grad_op = Roll(scope, grad_inputs[0], Neg(scope, shift), axis);
      grad_outputs->push_back(grad_op);
      grad_outputs->push_back(NoGradient());
      grad_outputs->push_back(NoGradient());
      return scope.status();
    }
    REGISTER_GRADIENT_OP("Roll", RollGrad);
    
    }  // namespace
    }  // namespace ops
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 19 12:19:42 UTC 2020
    - 1.4K bytes
    - Viewed (0)
  4. tensorflow/cc/gradients/functional_grad.cc

      for (int i = 0; i < num_inputs; i++) {
        func_inputs.push_back(op.input(i));
        input_dtypes.push_back(op.input_type(i));
      }
    
      func_inputs.insert(std::end(func_inputs), std::begin(grad_inputs),
                         std::end(grad_inputs));
    
      auto grad = SymbolicGradient(scope, func_inputs, input_dtypes, f);
      for (int i = 0; i < num_inputs; i++) {
        grad_outputs->push_back(grad[i]);
      }
    
      return scope.status();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Oct 15 20:09:06 UTC 2021
    - 2.1K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/ops/gen/model/op_spec.cc

        output_args_.push_back(arg);
      }
      // Parse the attributes.
      for (const OpDef::AttrDef& attr_def : op_def.attr()) {
        AttrSpec attr = AttrSpec::Create(attr_def);
        // Only non-inferred args are added as arguments.
        if (inferred_attrs.find(attr_def.name()) == inferred_attrs.end()) {
          argument_attrs_.push_back(attr);
        }
      }
    }
    
    }  // namespace generator
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 01 21:05:56 UTC 2021
    - 2.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/compilation_passes_test_main.cc

      args.reserve(real_argc + 1);
      for (int i = 0; i < real_argc; i++) {
        args.push_back(real_argv[i]);
      }
    
      struct FreeDeleter {
        void operator()(char* ptr) { free(ptr); }
      };
    
      std::unique_ptr<char, FreeDeleter> enable_global_jit_arg(
          strdup("--tf_xla_cpu_global_jit=true"));
      args.push_back(enable_global_jit_arg.get());
    
      std::unique_ptr<char, FreeDeleter> reduce_min_cluster_size_arg(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 14 00:31:28 UTC 2019
    - 1.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/decompose_hybrid_quantization.cc

            auto newTy = QuantizedType::castToExpressedType(operand.getType());
            newOperands.push_back(
                rewriter.create<TFL::DequantizeOp>(loc, newTy, operand));
            continue;
          }
    
          newOperands.push_back(operand);
        }
    
        SmallVector<Type> newResultTys;
        for (auto result : op->getResults()) {
          Type resultTy = result.getType();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.8K bytes
    - Viewed (0)
  8. tensorflow/c/experimental/saved_model/core/test_utils.cc

      result.reserve(set.size());
      for (DataType dt : set) {
        result.push_back(dt);
      }
      return result;
    }
    
    std::vector<std::vector<int64_t>> InterestingShapes() {
      std::vector<std::vector<int64_t>> interesting_shapes;
      interesting_shapes.push_back({});             // Scalar
      interesting_shapes.push_back({10});           // 1D Vector
      interesting_shapes.push_back({3, 3});         // 2D Matrix
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Sep 08 20:13:32 UTC 2023
    - 5.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc

        if (aggregator_ops_to_ignore.count(op.getId().str())) continue;
    
        ids.push_back(op.getId());
        calibration_methods.push_back(op.getCalibrationMethod());
        statistics_outputs.push_back(op.getMin());
        statistics_outputs.push_back(op.getMax());
        statistics_outputs.push_back(op.getHistogram());
      }
    }
    
    // Inserts a `CalibrationStatisticsSaverOp` to the end of the region.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/python/mlir_wrapper/basic_classes.cc

          .def("front", &mlir::Region::front, py::return_value_policy::reference)
          .def("add_block", [](mlir::Region& r) { r.push_back(new mlir::Block); })
          .def("push_back", &mlir::Region::push_back)
          .def("size", [](mlir::Region& r) { return r.getBlocks().size(); })
          .def("front", &mlir::Region::front, py::return_value_policy::reference);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jan 22 04:26:07 UTC 2022
    - 2.4K bytes
    - Viewed (0)
Back to top