| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | extract_image_patches_op.cc | 78 int input_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in Compile() local
111 int input_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in Compile() local
|
| D | conv_op_helpers.cc | 124 int input_dim = GetTensorSpatialDimIndex(num_dims, attrs.data_format, i); in CheckConvAttrs() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | batch_to_space_nd_test.cc | 25 int input_dim, int output_dim) { in GetIndexRange()
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | model_subclassing_test_util.py | 94 def get_functional_graph_model(input_dim, num_classes): argument
109 def get_nested_model_3(input_dim, num_classes): argument
|
| D | model_architectures.py | 159 def __init__(self, input_dim=3): argument
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | reshape_tester.h | 35 for (int32_t input_dim : input_shape) { in InputShape() local
|
| /external/tensorflow/tensorflow/python/debug/examples/v1/ |
| D | debug_mnist_v1.py | 162 def nn_layer(input_tensor, input_dim, output_dim, layer_name, act=tf.nn.relu): argument
|
| /external/tensorflow/tensorflow/python/debug/examples/v2/ |
| D | debug_mnist_v2.py | 164 def get_dense_weights(input_dim, output_dim): argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | embeddings.py | 92 input_dim, argument
|
| D | kernelized.py | 274 def _get_default_scale(initializer, input_dim): argument
|
| D | recurrent_test.py | 67 def __init__(self, units, input_dim): argument
107 def __init__(self, units, input_dim): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | optimizers_test.py | 38 def _get_model(input_dim, num_hidden, output_dim): argument
|
| D | testing_utils.py | 450 def get_small_sequential_mlp(num_hidden, num_classes, input_dim=None): argument
461 def get_small_functional_mlp(num_hidden, num_classes, input_dim): argument
522 def get_small_mlp(num_hidden, num_classes, input_dim): argument
|
| /external/XNNPACK/src/operators/ |
| D | constant-pad-nd.c | 125 const size_t input_dim = input_shape[num_dims - 1 - i]; in setup_constant_pad_nd() local
|
| /external/XNNPACK/test/ |
| D | constant-pad-operator-tester.h | 35 inline size_t input_dim(size_t i) const { in input_dim() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder.cc | 341 HloInstruction* reshape, int64 input_dim, in RewriteDynamicReshapeSplitInput()
571 for (int64 input_dim : input_dims) { in RewriteDynamicReshapeCombineInput() local
591 const int64 input_dim = input_dims[i]; in RewriteDynamicReshapeCombineInput() local
718 int64 input_dim = input_dims[0]; in RewriteDynamicReshapeSingleGroup() local
1893 for (int64 input_dim = 0; input_dim < operand->shape().rank(); in Run() local
|
| D | shape_inference.cc | 2975 for (int64 input_dim = 0; input_dim < operand.rank(); ++input_dim) { in InferReshapeShape() local
|
| D | hlo_evaluator.cc | 1670 int64 input_dim_value_to_output_index(int64 input_dim) { in input_dim_value_to_output_index()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorConvolution.h | 33 const Index input_dim = input_dims[index]; in IndexMapper() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 190 const nvinfer1::Dims input_dim = input->getDimensions(); in GetNetworkInputShapes() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | depthwise_conv_op_test.py | 115 def PaddingsForDim(input_dim, filter_dim, stride): argument
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 203 const int64 input_dim = input.dim_size(axis); in RecordLabelToDimension() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1389 for (int64 input_dim = 0; input_dim < input_shape.rank(); ++input_dim) { in ReshapeIsBitcast() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2287 const int64_t input_dim = input_type.getDimSize(i); in Verify() local
|
| D | tf_ops_a_m.cc | 363 int64_t input_dim = input_shape[spatial_dim_index]; in Verify() local
|