| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 36 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
56 const TensorShape input_shape = ctx->InputShape(i); in Compile() local
75 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
91 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
110 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
164 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
222 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
236 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/layers/ |
| D | advanced_activations.py | 67 def compute_output_shape(self, input_shape): argument
122 def build(self, input_shape): argument
165 def compute_output_shape(self, input_shape): argument
204 def compute_output_shape(self, input_shape): argument
243 def compute_output_shape(self, input_shape): argument
277 def compute_output_shape(self, input_shape): argument
|
| D | merge.py | 84 def build(self, input_shape): argument
178 def compute_output_shape(self, input_shape): argument
271 def build(self, input_shape): argument
367 def build(self, input_shape): argument
389 def compute_output_shape(self, input_shape): argument
475 def build(self, input_shape): argument
520 def compute_output_shape(self, input_shape): argument
|
| D | local.py | 124 def build(self, input_shape): argument
152 def compute_output_shape(self, input_shape): argument
311 def build(self, input_shape): argument
354 def compute_output_shape(self, input_shape): argument
|
| D | core.py | 85 def compute_output_shape(self, input_shape): argument
310 def compute_output_shape(self, input_shape): argument
359 def _fix_unknown_dimension(self, input_shape, output_shape): argument
404 def compute_output_shape(self, input_shape): argument
461 def compute_output_shape(self, input_shape): argument
529 def compute_output_shape(self, input_shape): argument
611 def _compute_output_shape(self, input_shape): argument
853 def compute_output_shape(self, input_shape): argument
|
| D | embeddings.py | 115 def build(self, input_shape): argument
132 def compute_output_shape(self, input_shape): argument
|
| D | noise.py | 71 def compute_output_shape(self, input_shape): argument
118 def compute_output_shape(self, input_shape): argument
190 def compute_output_shape(self, input_shape): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | nth_element_op_test.py | 61 def _testFloatLargeInput(self, input_shape): argument
72 def _testIntLargeInput(self, input_shape): argument
83 def _testLargeInput(self, input_shape): argument
96 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/keras/_impl/keras/ |
| D | testing_utils.py | 30 input_shape, argument
54 def layer_test(layer_cls, kwargs=None, input_shape=None, input_dtype=None, argument
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/applications/ |
| D | densenet.py | 131 input_shape=None, argument
307 input_shape=None, argument
319 input_shape=None, argument
331 input_shape=None, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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/contrib/lite/toco/graph_transformations/ |
| D | resolve_constant_strided_slice.cc | 26 int StartForAxis(StridedSliceOperator const& op, Shape const& input_shape, in StartForAxis()
43 int StopForAxis(StridedSliceOperator const& op, Shape const& input_shape, in StopForAxis()
86 Shape const& input_shape = input_array.shape(); in StridedSlice() local
|
| 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()
121 const Shape input_shape = input_array.shape(); in Run() local
|
| D | propagate_fixed_sizes.cc | 32 void ComputeConvSizes(const Shape& input_shape, int output_depth, int kwidth, in ComputeConvSizes()
153 const auto& input_shape = input_array.shape(); in ProcessConvOperator() local
195 const auto& input_shape = input_array.shape(); in ProcessDepthwiseConvOperator() local
236 const auto& input_shape = input_array.shape(); in ProcessDepthToSpaceOperator() local
259 const auto& input_shape = input_array.shape(); in ProcessSpaceToDepthOperator() local
314 const auto& input_shape = input_array.shape(); in ProcessFullyConnectedOperator() local
348 const auto& input_shape = input_array.shape(); in ProcessTensorFlowReshapeOperator() local
476 const auto& input_shape = input_array.shape(); in ProcessTensorFlowReductionOperator() local
522 const Shape& input_shape = input_array.shape(); in ProcessSliceOperator() local
549 const auto& input_shape = input_array.shape(); in ProcessReorderAxesOperator() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops.cc | 48 StringPiece label, const TensorShape& input_shape, in ConvBackpropExtractAndVerifyDimensionV2()
87 StringPiece label, const TensorShape& input_shape, in ConvBackpropExtractAndVerifyDimension()
98 StringPiece label, int num_spatial_dims, const TensorShape& input_shape, in ConvBackpropComputeDimensionsV2()
151 const TensorShape& input_shape, in ConvBackpropComputeDimensions()
|
| D | reduce_join_op.cc | 51 const TensorShape& input_shape, in LinearSubIndexToFullIndex()
66 const TensorShape& input_shape) { in GetReductionIterSize()
100 const TensorShape& input_shape, bool keep_dims) { in GetOutputShape()
126 const TensorShape& input_shape = input.shape(); in Compute() local
|
| /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
|
| D | conv2d_benchmark.py | 34 def build_graph(device, input_shape, filter_shape, strides, padding, dtype, argument
84 def _run_graph(self, device, input_shape, filter_shape, strides, padding, argument
|
| /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/compiler/xla/tests/ |
| D | convolution_test.cc | 125 Shape input_shape = MakeShapeWrapper<T>({1, 1, 1, 2}); in RunTest() local
156 Shape input_shape = MakeShapeWrapper<T>({1, 1, 4, 4}); in RunTest() local
190 Shape input_shape = MakeShapeWrapper<T>({1, 1, 4, 4}); in RunTest() local
226 Shape input_shape = MakeShapeWrapper<T>({1, 1, 4, 4}); in RunTest() local
254 Shape input_shape = ShapeUtil::MakeShape(F32, {1, 2, 5}); in XLA_TEST_F() local
284 Shape input_shape = MakeShapeWrapper<T>({1, 2, 5}); in RunTest() local
320 Shape input_shape = ShapeUtil::MakeShape(F32, {1, 2, 5}); in XLA_TEST_F() local
351 Shape input_shape = ShapeUtil::MakeShape(F32, {1, 2, 5}); in XLA_TEST_F() local
385 Shape input_shape = MakeShapeWrapper<T>({1, 2, 5}); in RunTest() local
423 Shape input_shape = ShapeUtil::MakeShape(F32, input_dims); in XLA_TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | chain.py | 118 def _shape_helper(self, func_name, input_shape, reverse): argument
128 def _forward_event_shape(self, input_shape): argument
132 def _forward_event_shape_tensor(self, input_shape): argument
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| D | core_layers.py | 124 def build(self, input_shape): argument
213 def compute_output_shape(self, input_shape): argument
398 def build(self, input_shape): argument
470 def compute_output_shape(self, input_shape): argument
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | gru_ops.py | 177 def build(self, input_shape): argument
222 def build(self, input_shape): argument
|