| /external/tensorflow/tensorflow/contrib/kernel_methods/python/mappers/ |
| D | random_fourier_features.py | 60 def __init__(self, input_dim, output_dim, stddev=1.0, seed=1, name=None): argument
100 def input_dim(self): member in RandomFourierFeatureMapper
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | extract_image_patches_op.cc | 71 int input_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in Compile() local
104 int input_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in Compile() local
|
| D | conv_op_helpers.cc | 190 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/examples/tutorials/mnist/ |
| D | mnist_with_summaries.py | 78 def nn_layer(input_tensor, input_dim, output_dim, layer_name, act=tf.nn.relu): argument
|
| /external/tensorflow/tensorflow/python/debug/examples/ |
| D | debug_mnist.py | 77 def nn_layer(input_tensor, input_dim, output_dim, layer_name, act=tf.nn.relu): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | testing_utils.py | 288 def get_small_sequential_mlp(num_hidden, num_classes, input_dim=None): argument
300 def get_small_functional_mlp(num_hidden, num_classes, input_dim): argument
350 def get_small_mlp(num_hidden, num_classes, input_dim): argument
|
| D | optimizers_test.py | 34 def _get_model(input_dim, num_hidden, output_dim): argument
|
| D | model_subclassing_test.py | 135 def get_functional_graph_model(input_dim, num_classes): argument
163 def get_nested_model_3(input_dim, num_classes): argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | embeddings.py | 92 input_dim, argument
|
| D | kernelized.py | 254 def _get_default_scale(initializer, input_dim): argument
|
| D | recurrent_test.py | 64 def __init__(self, units, input_dim): argument
104 def __init__(self, units, input_dim): argument
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorConvolution.h | 33 const Index input_dim = input_dims[index]; in IndexMapper() local
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op_test_base.py | 616 def _CalculateConvolvedOutputDim(input_dim, filter_dim, stride, padding_type): argument
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1231 for (int64 input_dim = 0; input_dim < input_shape.rank(); ++input_dim) { in ReshapeIsBitcast() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1000 int64 input_dim_value_to_output_index(int64 input_dim) { in input_dim_value_to_output_index()
|
| D | hlo_evaluator_typed_visitor.h | 2278 int64 input_dim_value_to_update_index(int64 input_dim) { in input_dim_value_to_update_index()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3497 for (int input_dim : input_dims) { in GetDimensionsToKeepMinorToMajor() local
3538 for (int input_dim : input_dims) { in GetReductionToVectorDimensions() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3805 auto input_dim = tensor->getDimensions(); in ConvertMatMulHelper() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5765 int input_dim, int output_dim, int* start_index, in GetIndexRange()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 6093 int input_dim, int output_dim, int* start_index, in GetIndexRange()
|