| /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
104 def output_dim(self): member in RandomFourierFeatureMapper
|
| D | dense_kernel_mapper.py | 56 def output_dim(self): member in DenseKernelMapper
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | kernelized_test.py | 180 def test_compute_output_shape(self, output_dim, initializer, scale): argument
209 def test_get_config(self, output_dim, initializer, scale, trainable): argument
238 def test_from_config(self, output_dim, initializer, scale, trainable): argument
271 def test_same_random_features_params_reused(self, output_dim, initializer, argument
|
| D | embeddings.py | 93 output_dim, argument
|
| D | kernelized.py | 132 output_dim, argument
|
| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | applications_test.py | 46 def test_feature_extration_model(self, model_fn, output_dim): 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/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/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 215 for (int64 output_dim : output_dims) { in FoldGatherOfGather() local
537 for (int64 output_dim : operand->output_dims()) { in ReshapeToAddDegenerateDims() local
564 for (int64 output_dim : new_output_dims) { in ReshapeToAddDegenerateDims() local
711 int64 output_dim = scalar_indexed->output_dims()[i]; in FoldReshapeOfGatherNoDegenerateDims() local
877 auto is_broadcasted_dim = [&](int64 output_dim) { in ComputeArrayForElementwiseBinaryOp()
|
| D | layout_assignment.cc | 1040 int64 output_dim = LayoutUtil::Minor(output_layout, i); in ChooseOperandLayoutFromOutputLayout() local
|
| D | hlo_evaluator.cc | 1097 int64 output_dim = in HandleGather() local
|
| D | elemental_ir_emitter.cc | 1916 int64 output_dim = operand_to_output_dim[operand_dim]; in EmitElementalGather() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | optimizers_test.py | 34 def _get_model(input_dim, num_hidden, output_dim): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops.cc | 167 DimensionHandle output_dim; in CombinedNMSShapeFn() local
855 DimensionHandle output_dim; in __anon6a71d27f1802() local
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | rnn_cell.py | 3441 output_dim=None, argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3820 auto output_dim = output_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()
|