Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 164 for pushBack (0.23 sec)

  1. tensorflow/compiler/jit/device_util.cc

        return it->second;
      }
    
      int new_id = names_.size();
      names_.push_back(string(name));
      id_to_device_type_.push_back(std::make_unique<DeviceType>(""));
      DeviceType* device_type = id_to_device_type_.back().get();
      TF_RETURN_IF_ERROR(DeviceNameToDeviceType(names_.back(), device_type));
    
      is_cpu_.push_back(device_type->type_string() == DEVICE_CPU);
      is_gpu_.push_back(device_type->type_string() == DEVICE_GPU);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. tensorflow/compiler/mlir/tensorflow/utils/topological_sort.cc

            remaining_incoming_ctrl_edges[successor]--;
            if (remaining_incoming_ctrl_edges[successor] == 0 &&
                remaining_incoming_data_edges[successor] == 0) {
              ready.push_back(successor);
            }
          }
        }
        result.push_back(best);
      }
      return result;
    }
    
    }  // namespace TF
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 08 17:01:11 UTC 2022
    - 5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/ir/tf_structs.cc

    #include <optional>
    
    namespace mlir {
    namespace TF {
    
    void RuntimeDevices::AddDevice(const ParsedName& device) {
      device_names_.push_back(device);
    }
    
    void RuntimeDevices::AddGpuDevice(const ParsedName& device,
                                      const GpuDeviceMetadata& metadata) {
      device_names_.push_back(device);
      gpu_metadata_.insert({DeviceNameUtils::ParsedNameToString(device), metadata});
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 02 20:41:19 UTC 2023
    - 1.4K bytes
    - Viewed (0)
  9. tensorflow/c/experimental/ops/gen/common/case_format.cc

          if (wordStart) {
            result.push_back(delimiter);
          }
          wordStart = true;
          continue;
        }
        if (!from_snake && isupper(c)) {
          wordStart = true;
        }
    
        // add delimiter
        if (wordStart && toSnake && !inputStart) {
          result.push_back(delimiter);
        }
    
        // add the next letter from the input string (choosing upper/lower case)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 09:51:28 UTC 2024
    - 3K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/group_by_dialect.cc

      std::vector<Type> input_types;
      std::vector<Type> output_types;
    
      input_types.reserve(inputs.size());
      for (Value v : inputs) {
        input_types.push_back(v.getType());
      }
      output_types.reserve(outputs.size());
      for (Value v : outputs) {
        output_types.push_back(v.getType());
      }
    
      // Create the function.
      MLIRContext* context = ops[0]->getContext();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 17 07:31:01 UTC 2023
    - 8K bytes
    - Viewed (0)
Back to top