| /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/python/keras/tests/ |
| D | model_subclassing_test_util.py | 90 def get_functional_graph_model(input_dim, num_classes): argument
105 def get_nested_model_3(input_dim, num_classes): argument
|
| D | model_architectures.py | 156 def __init__(self, input_dim=3): argument
|
| /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/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 | 107 input_dim, argument
|
| D | kernelized.py | 269 def _get_default_scale(initializer, input_dim): argument
|
| D | recurrent_test.py | 63 def __init__(self, units, input_dim): argument
103 def __init__(self, units, input_dim): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_elemental_utils.cc | 137 auto input_dim = b->create<AddIOp>( in elementalLower() local
167 int input_dim = std::distance(broadcast_dimensions.begin(), it); in elementalLowerImplForBroadcastInDimOps() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | optimizers_test.py | 34 def _get_model(input_dim, num_hidden, output_dim): argument
|
| D | testing_utils.py | 422 def get_small_sequential_mlp(num_hidden, num_classes, input_dim=None): argument
433 def get_small_functional_mlp(num_hidden, num_classes, input_dim): argument
499 def get_small_mlp(num_hidden, num_classes, input_dim): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder.cc | 351 HloInstruction* reshape, int64_t input_dim, in RewriteDynamicReshapeSplitInput()
581 for (int64_t input_dim : input_dims) { in RewriteDynamicReshapeCombineInput() local
601 const int64_t input_dim = input_dims[i]; in RewriteDynamicReshapeCombineInput() local
728 int64_t input_dim = input_dims[0]; in RewriteDynamicReshapeSingleGroup() local
2107 for (int64_t input_dim = 0; input_dim < operand->shape().rank(); in Run() local
|
| D | shape_inference.cc | 3112 for (int64_t input_dim = 0; input_dim < operand.rank(); ++input_dim) { in InferReshapeShape() local
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorConvolution.h | 33 const Index input_dim = input_dims[index]; in IndexMapper() local
|
| D | TensorDeviceSycl.h | 546 const std::array<Index, 2> &input_dim, cl::sycl::range<2> &global_range, in parallel_for_setup()
585 const std::array<Index, 3> &input_dim, cl::sycl::range<3> &global_range, in parallel_for_setup()
|
| /external/XNNPACK/src/operators/ |
| D | constant-pad-nd.c | 147 const size_t input_dim = input_shape[num_dims - 1 - i]; in setup_constant_pad_nd() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 176 const nvinfer1::Dims input_dim = input->getDimensions(); in GetNetworkInputShapes() 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/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_t input_dim = input.dim_size(axis); in RecordLabelToDimension() local
|
| /external/armnn/src/armnnCaffeParser/ |
| D | RecordByRecordCaffeParser.cpp | 553 int input_dim = ReadBase128(ifs); in CreateNetworkFromBinaryFile() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1429 for (int64_t input_dim = 0; input_dim < input_shape.rank(); ++input_dim) { in ReshapeIsBitcast() local
|