| /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/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 | 158 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 | 160 def get_dense_weights(input_dim, output_dim): argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | embeddings.py | 107 input_dim, argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| 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/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/xla/service/ |
| D | dynamic_padder.cc | 316 HloInstruction* reshape, int64_t input_dim, in GenerateBinaryMask()
469 HloInstruction* reshape, int64_t input_dim, in RewriteDynamicReshapeSplitInput()
783 int64_t input_dim = input_dims[0]; in RewriteDynamicReshapeSingleGroup() local
2262 for (int64_t input_dim = 0; input_dim < operand->shape().rank(); in Run() local
|
| D | hlo_evaluator.cc | 2416 int64_t input_dim_value_to_output_index(int64_t input_dim) { in input_dim_value_to_output_index()
2809 int64_t input_dim_value_to_update_index(int64_t input_dim) { in input_dim_value_to_update_index()
|
| D | shape_inference.cc | 3115 for (int64_t input_dim = 0; input_dim < operand.rank(); ++input_dim) { in InferReshapeShape() local
|
| /external/XNNPACK/test/ |
| D | constant-pad-operator-tester.h | 34 inline size_t input_dim(size_t i) const { in input_dim() function
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 83 const int64_t input_dim = input.dim_size(axis); in RecordLabelToDimension() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | depthwise_conv_op_base.py | 112 def PaddingsForDim(input_dim, filter_dim, stride): argument
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1471 for (int64_t input_dim = 0; input_dim < input_shape.rank(); ++input_dim) { in ReshapeIsBitcast() local
1567 const int64_t input_dim = input_shape.layout().minor_to_major(input_minor); in AlignLayouts() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 411 int64_t input_dim = input_shape[spatial_dim_index]; in verify() local
|
| D | tf_ops_n_z.cc | 2469 const int64_t input_dim = input_type.getDimSize(i); in verify() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4497 nvinfer1::Dims input_dim = input_a.GetTrtDims(); in ConvertFullyConnectedImpl() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3859 int64_t input_dim = (*(input_shape_attr.begin() + i)).getSExtValue(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4183 int input_dim, int output_dim, int* start_index, in GetIndexRange()
|