OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:input_batch_size
(Results
1 - 5
of
5
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
batchtospace_op.cc
139
int64
input_batch_size
= orig_input_batch_size;
local
142
input_batch_size
*= size;
145
internal_input_shape.AddDim(
input_batch_size
);
146
internal_output_shape.AddDim(
input_batch_size
/ block_shape_product);
spacetobatch_op.cc
136
int64
input_batch_size
= orig_input_tensor.dim_size(0);
local
139
input_batch_size
*= size;
142
internal_input_shape.AddDim(
input_batch_size
);
143
internal_output_shape.AddDim(
input_batch_size
* block_shape_product);
/external/tensorflow/tensorflow/contrib/training/python/training/
sampling_ops_test.py
186
input_batch_size
= 11
187
val_input_batch = [array_ops.zeros([
input_batch_size
, 2, 3, 4])]
190
lambda: array_ops.ones([
input_batch_size
], dtype=dtypes.int32) * 1,
191
lambda: array_ops.ones([
input_batch_size
], dtype=dtypes.int32) * 3)
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
reference_ops.h
[
all
...]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
optimized_ops.h
[
all
...]
Completed in 318 milliseconds