| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | conv_utils_test.py | 30 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):
|
| D | tf_utils.py | 181 def convert_shapes(input_shape, to_tuples=True): argument
207 def _is_atomic_shape(input_shape): argument
218 def _convert_shape(input_shape): argument
288 def wrapper(instance, input_shape): argument
|
| D | conv_utils.py | 230 def conv_kernel_mask(input_shape, kernel_shape, strides, padding): argument
309 def conv_connected_inputs(input_shape, argument
368 def conv_output_shape(input_shape, kernel_shape, strides, padding): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 39 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
59 const TensorShape input_shape = ctx->InputShape(i); in Compile() local
78 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
94 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
119 const TensorShape input_shape = ctx->InputShape("input"); in Compile() local
169 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
226 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
240 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
|
| D | diag_op.cc | 86 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
126 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
170 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
197 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
|
| D | dynamic_slice_ops.cc | 40 const TensorShape input_shape = ctx->InputShape("input"); in Compile() local
75 const TensorShape input_shape = ctx->InputShape("input"); in Compile() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | resize_nearest_neighbor_test.cc | 30 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/contrib/model_pruning/python/ |
| D | pruning_utils_test.py | 83 def testFactorizedAvgPool(self, input_shape, window_shape): argument
93 def testFactorizedMaxPool(self, input_shape, window_shape): argument
103 def testExpandTensor(self, input_shape, block_dim): argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | advanced_activations.py | 67 def compute_output_shape(self, input_shape): argument
121 def build(self, input_shape): argument
157 def compute_output_shape(self, input_shape): argument
195 def compute_output_shape(self, input_shape): argument
234 def compute_output_shape(self, input_shape): argument
268 def compute_output_shape(self, input_shape): argument
331 def compute_output_shape(self, input_shape): argument
|
| D | core.py | 100 def compute_output_shape(self, input_shape): argument
164 def compute_output_shape(self, input_shape): argument
361 def compute_output_shape(self, input_shape): argument
410 def _fix_unknown_dimension(self, input_shape, output_shape): argument
452 def compute_output_shape(self, input_shape): argument
515 def compute_output_shape(self, input_shape): argument
585 def compute_output_shape(self, input_shape): argument
633 def compute_output_shape(self, input_shape): argument
742 def compute_output_shape(self, input_shape): argument
979 def build(self, input_shape): argument
[all …]
|
| D | local.py | 159 def build(self, input_shape): argument
226 def compute_output_shape(self, input_shape): argument
434 def build(self, input_shape): argument
509 def compute_output_shape(self, input_shape): argument
584 def get_locallyconnected_mask(input_shape, argument
|
| D | merge.py | 87 def build(self, input_shape): argument
185 def compute_output_shape(self, input_shape): argument
279 def build(self, input_shape): argument
376 def build(self, input_shape): argument
398 def compute_output_shape(self, input_shape): argument
484 def build(self, input_shape): argument
529 def compute_output_shape(self, input_shape): argument
|
| D | wrappers.py | 58 def build(self, input_shape=None): argument
191 def build(self, input_shape): argument
209 def compute_output_shape(self, input_shape): argument
415 def compute_output_shape(self, input_shape): argument
580 def build(self, input_shape): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference_test.cc | 82 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
105 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
128 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
157 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
180 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
212 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
338 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 3}); in TEST_F() local
377 auto input_shape = ShapeUtil::MakeShape(F32, {2, 3, 4, 5, 6}); in TEST_F() local
411 auto input_shape = ShapeUtil::MakeShape(F32, {2, 3, 4, 5, 6}); in TEST_F() local
436 auto input_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
[all …]
|
| D | tuple_util.cc | 27 const Shape& input_shape = input_tuple->shape(); in ExtractPrefix() local
47 const Shape& input_shape = input_tuple->shape(); in AppendSuffix() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops.cc | 83 auto input_shape = c->input(0); in __anone1d60d010302() local
126 auto input_shape = c->input(0); in __anone1d60d010402() local
172 auto input_shape = c->input(0); in __anone1d60d010502() local
222 auto input_shape = c->input(0); in __anone1d60d010602() local
259 auto input_shape = c->input(0); in __anone1d60d010702() local
298 auto input_shape = c->input(0); in __anone1d60d010802() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | nth_element_op_test.py | 62 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/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_benchmark.py | 32 def build_conv_bias_relu_graph(device, input_shape, filter_shape, strides, argument
77 def build_fused_conv_bias_relu_graph(device, input_shape, filter_shape, strides, argument
134 def _run_graph(self, device, input_shape, filter_shape, strides, padding, argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | transpose_benchmark.py | 34 def build_graph(device, input_shape, perm, datatype, num_iters): argument
65 def _run_graph(self, device, input_shape, perm, num_iters, datatype): argument
|
| D | concat_benchmark.py | 35 def build_graph(device, input_shape, variable, num_inputs, axis, grad): argument
81 def _run_graph(self, device, input_shape, variable, num_inputs, axis, grad, argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test.cc | 117 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/contrib/specs/python/ |
| D | summaries.py | 247 input_shape=None, argument
275 input_shape=None, argument
299 input_shape=None, argument
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_reorder_axes.cc | 30 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
|
| D | propagate_fixed_sizes.cc | 34 void ComputeConvSizes(const Shape& input_shape, int output_depth, int kwidth, in ComputeConvSizes()
130 const auto& input_shape = input_array.shape(); in ProcessConvOperator() local
229 const auto& input_shape = input_array.shape(); in ProcessTransposeConvOperator() local
259 const auto& input_shape = input_array.shape(); in ProcessDepthwiseConvOperator() local
301 const auto& input_shape = input_array.shape(); in ProcessDepthToSpaceOperator() local
324 const auto& input_shape = input_array.shape(); in ProcessSpaceToDepthOperator() local
374 const auto& input_shape = input_array.shape(); in ProcessFullyConnectedOperator() local
408 const auto& input_shape = input_array.shape(); in ProcessTensorFlowReshapeOperator() local
562 const auto& input_shape = input_array.shape(); in ProcessTensorFlowReductionOperator() local
620 const Shape& input_shape = input_array.shape(); in ProcessSliceOperator() local
[all …]
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | chain.py | 224 def _shape_helper(self, func_name, input_shape, reverse): argument
234 def _forward_event_shape(self, input_shape): argument
238 def _forward_event_shape_tensor(self, input_shape): argument
|