OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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
...]