HomeSort by relevance Sort by last modified time
    Searched refs:tensor_size (Results 1 - 25 of 32) sorted by null

1 2

  /external/tensorflow/tensorflow/lite/tools/optimize/calibration/
calibration_logger.h 27 void Update(const float* values, size_t tensor_size) {
29 if (tensor_size <= 0) return;
39 for (size_t i = 0; i < tensor_size; i++) {
68 size_t tensor_size) {
69 tensor_id_to_stats_map_[tensor_index].Update(tensor_values, tensor_size);
calibrator_test.cc 71 const size_t tensor_size = 1 * 8 * 8 * 3; local
73 std::vector<float> ones(tensor_size, 1.0f);
80 ASSERT_EQ(tensor->bytes, tensor_size * sizeof(float));
81 for (size_t j = 0; j < tensor_size; j++) {
91 for (size_t i = 0; i < tensor_size; i++) {
95 for (size_t i = 0; i < tensor_size; i++) {
134 const size_t tensor_size = 1 * 8 * 8 * 3; local
141 ASSERT_EQ(tensor->bytes, tensor_size * sizeof(float));
142 for (size_t j = 0; j < tensor_size; j++) {
  /external/tensorflow/tensorflow/core/grappler/inputs/
trivial_test_graph_input_yielder.h 31 TrivialTestGraphInputYielder(int num_stages, int width, int tensor_size,
trivial_test_graph_input_yielder.cc 31 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size,
39 const int batch_size = tensor_size < 0 ? 1 : tensor_size;
90 int num_stages, int width, int tensor_size, bool insert_queue,
94 tensor_size_(tensor_size),
  /external/tensorflow/tensorflow/core/distributed_runtime/
rpcbench_test.cc 124 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size,
133 Output x = Const(s.WithOpName("x"), 0.0f, {tensor_size, 1});
157 string DebugString(const Tensor& x, const Tensor& y, int tensor_size) {
158 CHECK_EQ(x.NumElements(), tensor_size);
159 CHECK_EQ(y.NumElements(), tensor_size);
163 CHECK_GE(tensor_size, 2);
169 static void BM_Helper(int iters, int width, int num_stages, int tensor_size,
176 GraphDef def = CreateGraphDef(num_stages, width, tensor_size,
183 Tensor x(DT_FLOAT, TensorShape({tensor_size, 1}));
188 "; tensor bytes/send: ", tensor_size * sizeof(float)))
    [all...]