Home
last modified time | relevance | path

Searched defs:input_shape (Results 1 – 25 of 425) sorted by relevance

12345678910>>...17

/external/tensorflow/tensorflow/python/keras/utils/
Dconv_utils_test.py30 def _get_const_output_shape(input_shape, dim): argument
166 def test_conv_kernel_mask_fc(self, *input_shape):
183 def test_conv_kernel_mask_diag(self, *input_shape):
201 def test_conv_kernel_mask_full_stride(self, *input_shape):
222 def test_conv_kernel_mask_almost_full_stride(self, *input_shape):
245 def test_conv_kernel_mask_rect_kernel(self, *input_shape):
278 def test_conv_kernel_mask_wrong_padding(self, *input_shape):
301 def test_conv_kernel_mask_wrong_dims(self, *input_shape):
Dtf_utils.py187 def convert_shapes(input_shape, to_tuples=True): argument
213 def _is_atomic_shape(input_shape): argument
224 def _convert_shape(input_shape): argument
301 def wrapper(instance, input_shape): argument
Dconv_utils.py236 def conv_kernel_mask(input_shape, kernel_shape, strides, padding): argument
312 def conv_kernel_idxs(input_shape, kernel_shape, strides, padding, filters_in, argument
403 def conv_connected_inputs(input_shape, kernel_shape, output_position, strides, argument
459 def conv_output_shape(input_shape, kernel_shape, strides, padding): argument
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtranspose_utils_test.cc24 RuntimeShape input_shape({9}); in TEST() local
42 RuntimeShape input_shape({9, 3}); in TEST() local
62 RuntimeShape input_shape({9, 1}); in TEST() local
81 RuntimeShape input_shape({4, 3, 8}); in TEST() local
103 RuntimeShape input_shape({4, 1, 8}); in TEST() local
125 RuntimeShape input_shape({4, 1, 1}); in TEST() local
145 RuntimeShape input_shape({1, 1, 1}); in TEST() local
165 RuntimeShape input_shape({9, 3, 2, 4}); in TEST() local
189 RuntimeShape input_shape({9, 3, 1, 4}); in TEST() local
212 RuntimeShape input_shape({1, 3, 1, 4}); in TEST() local
[all …]
Dresize_nearest_neighbor_test.cc30 const RuntimeShape& input_shape, const std::vector<T>& input_data, in TestReferenceResizeNearestNeighbor()
48 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
59 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
70 RuntimeShape input_shape = {1, 3, 3, 1}; in TEST() local
81 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
92 RuntimeShape input_shape = {1, 4, 4, 1}; in TEST() local
104 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
115 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
132 RuntimeShape input_shape = {2, 2, 2, 2}; in TEST() local
158 RuntimeShape input_shape({batch, input_height, input_width, depth}); in TestOptimizedResizeNearestNeighbor() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dpad.h39 const RuntimeShape& input_shape, const T* input_data, in PadImpl()
112 const RuntimeShape& input_shape, const T* input_data, in Pad()
122 const RuntimeShape& input_shape, const T* input_data, in Pad()
133 const RuntimeShape& input_shape, const int32* input_data, in Pad()
145 const RuntimeShape& input_shape, const T* input_data, in PadImageStyle()
153 const RuntimeShape& input_shape, in PadImageStyle()
163 const RuntimeShape& input_shape, in PadImageStyle()
173 const RuntimeShape& input_shape, in PadImageStyle()
Dlogistic.h30 inline void Logistic(const RuntimeShape& input_shape, const float* input_data, in Logistic()
61 inline void Logistic(const LogisticParams&, const RuntimeShape& input_shape, in Logistic()
69 const RuntimeShape& input_shape, const int16* input_data, in Logistic()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dshape_op.cc42 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
76 const TensorShape input_shape = ctx->InputShape(i); in Compile() local
113 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
129 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
156 const TensorShape input_shape = ctx->InputShape("input"); in Compile() local
206 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
290 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
305 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
/external/tensorflow/tensorflow/python/keras/applications/
Defficientnet.py156 input_shape=None, argument
460 input_shape=None, argument
484 input_shape=None, argument
508 input_shape=None, argument
532 input_shape=None, argument
556 input_shape=None, argument
580 input_shape=None, argument
604 input_shape=None, argument
628 input_shape=None, argument
Dresnet_v2.py31 input_shape=None, argument
49 input_shape=None, argument
67 input_shape=None, argument
Ddensenet.py132 input_shape=None, argument
307 input_shape=None, argument
320 input_shape=None, argument
333 input_shape=None, argument
/external/tensorflow/tensorflow/python/keras/layers/
Dadvanced_activations.py81 def compute_output_shape(self, input_shape): argument
139 def build(self, input_shape): argument
175 def compute_output_shape(self, input_shape): argument
216 def compute_output_shape(self, input_shape): argument
258 def compute_output_shape(self, input_shape): argument
292 def compute_output_shape(self, input_shape): argument
378 def compute_output_shape(self, input_shape): argument
Dcore.py117 def compute_output_shape(self, input_shape): argument
186 def compute_output_shape(self, input_shape): argument
394 def compute_output_shape(self, input_shape): argument
447 def _fix_unknown_dimension(self, input_shape, output_shape): argument
489 def compute_output_shape(self, input_shape): argument
552 def compute_output_shape(self, input_shape): argument
641 def compute_output_shape(self, input_shape): argument
690 def compute_output_shape(self, input_shape): argument
816 def compute_output_shape(self, input_shape): argument
1110 def build(self, input_shape): argument
[all …]
Dmerge.py90 def build(self, input_shape): argument
186 def compute_output_shape(self, input_shape): argument
283 def build(self, input_shape): argument
490 def build(self, input_shape): argument
525 def compute_output_shape(self, input_shape): argument
657 def build(self, input_shape): argument
703 def compute_output_shape(self, input_shape): argument
Dlocal.py174 def build(self, input_shape): argument
262 def compute_output_shape(self, input_shape): argument
488 def build(self, input_shape): argument
584 def compute_output_shape(self, input_shape): argument
664 def get_locallyconnected_mask(input_shape, kernel_shape, strides, padding, argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dnth_element_op_test.py62 def _testFloatLargeInput(self, input_shape): argument
73 def _testIntLargeInput(self, input_shape): argument
84 def _testLargeInput(self, input_shape): argument
97 def _testEnumerateN(self, input_shape): argument
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dimage_preprocessing.py76 def build(self, input_shape): argument
89 def compute_output_shape(self, input_shape): argument
129 def build(self, input_shape): argument
162 def compute_output_shape(self, input_shape): argument
268 def compute_output_shape(self, input_shape): argument
309 def compute_output_shape(self, input_shape): argument
376 def compute_output_shape(self, input_shape): argument
497 def compute_output_shape(self, input_shape): argument
717 def compute_output_shape(self, input_shape): argument
839 def compute_output_shape(self, input_shape): argument
[all …]
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_integration_test.py128 def test_layer_output_type(self, layer_to_test, input_shape, _, layer_kwargs): argument
149 def _run_fit_eval_predict(self, layer_to_test, input_shape, data_shape, argument
192 def test_model_loops(self, layer_to_test, input_shape, fuzz_dims, argument
/external/tensorflow/tensorflow/core/ops/
Dcudnn_rnn_ops.cc84 auto input_shape = c->input(0); in __anon4fa58c430302() local
127 auto input_shape = c->input(0); in __anon4fa58c430402() local
174 auto input_shape = c->input(0); in __anon4fa58c430502() local
225 auto input_shape = c->input(0); in __anon4fa58c430602() local
262 auto input_shape = c->input(0); in __anon4fa58c430702() local
302 auto input_shape = c->input(0); in __anon4fa58c430802() local
/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/tests/
Dtranspose_test.cc70 const RuntimeShape input_shape = ::tflite::GetTensorShape(shape); in ComputeExpectedTransposeResult() local
89 std::vector<int> input_shape = {3}; in TEST() local
101 std::vector<int> input_shape = {3, 2}; in TEST() local
114 std::vector<int> input_shape = {2, 2, 3, 1}; in TEST() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_util.cc27 const Shape& input_shape = input_tuple->shape(); in ExtractPrefix() local
47 const Shape& input_shape = input_tuple->shape(); in AppendSuffix() local
Ddynamic_dimension_inference_test.cc102 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
125 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
148 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
177 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
200 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
232 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
265 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
438 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 3}); in TEST_F() local
477 auto input_shape = ShapeUtil::MakeShape(F32, {2, 3, 4, 5, 6}); in TEST_F() local
512 auto input_shape = ShapeUtil::MakeShape(F32, {4, 5}); in TEST_F() local
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_reorder_axes.cc30 Model* model, ReorderAxesOperator* reorder_op, const Shape& input_shape) { in CreateReshapeFromReorderAxes()
61 Model* model, ReorderAxesOperator* reorder_op, const Shape& input_shape, in CreateTransposeFromReorderAxes()
125 const Shape input_shape = input_array.shape(); in Run() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc117 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 1, 2}); in RunTest() local
148 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 4, 4}); in RunTest() local
182 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 4, 4}); in RunTest() local
218 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 4, 4}); in RunTest() local
246 Shape input_shape = ShapeUtil::MakeShape(F32, {1, 2, 5}); in XLA_TEST_F() local
276 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 5}); in RunTest() local
312 Shape input_shape = ShapeUtil::MakeShape(F32, {1, 2, 5}); in XLA_TEST_F() local
343 Shape input_shape = ShapeUtil::MakeShape(F32, {1, 2, 5}); in XLA_TEST_F() local
377 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 5}); in RunTest() local
415 Shape input_shape = ShapeUtil::MakeShape(F32, input_dims); in XLA_TEST_F() local
[all …]
/external/tensorflow/tensorflow/python/ops/
Dtranspose_benchmark.py34 def build_graph(device, input_shape, perm, datatype, num_iters): argument
65 def _run_graph(self, device, input_shape, perm, num_iters, datatype): argument

12345678910>>...17