| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | kernelized_test.py | 190 def test_compute_output_shape(self, output_dim, initializer, scale): argument
218 def test_get_config(self, output_dim, initializer, scale, trainable): argument
248 def test_from_config(self, output_dim, initializer, scale, trainable): argument
276 def test_same_random_features_params_reused(self, output_dim, initializer, argument
|
| D | embeddings.py | 108 output_dim, argument
|
| D | kernelized.py | 147 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/lite/delegates/xnnpack/ |
| D | reshape_tester.h | 46 for (int32_t output_dim : output_shape) { in OutputShape() local
|
| D | pad_tester.cc | 41 int32_t output_dim = InputShape()[i]; in OutputShape() local
|
| D | quantized_pad_tester.cc | 41 int32_t output_dim = InputShape()[i]; in OutputShape() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 215 for (int64_t output_dim : output_dims) { in FoldGatherOfGather() local
537 for (int64_t output_dim : operand->output_dims()) { in ReshapeToAddDegenerateDims() local
564 for (int64_t output_dim : new_output_dims) { in ReshapeToAddDegenerateDims() local
711 int64_t output_dim = scalar_indexed->output_dims()[i]; in FoldReshapeOfGatherNoDegenerateDims() local
877 auto is_broadcasted_dim = [&](int64_t output_dim) { in ComputeArrayForElementwiseBinaryOp()
|
| D | dynamic_padder.cc | 365 for (int64_t output_dim : output_dims) { in RewriteDynamicReshapeSplitInput() local
389 const int64_t output_dim = output_dims[i]; in RewriteDynamicReshapeSplitInput() local
482 for (int64_t output_dim : output_dims) { in RewriteDynamicReshapeSplitInput() local
567 int64_t output_dim, absl::Span<HloInstruction*> input_dynamic_dims, in RewriteDynamicReshapeCombineInput()
740 int64_t output_dim = output_dims[0]; in RewriteDynamicReshapeSingleGroup() local
|
| D | sharding_propagation.cc | 141 int64_t output_dim = in IsConvolutionKernelSmall() local
|
| D | layout_assignment.cc | 1183 int64_t output_dim = LayoutUtil::Minor(output_layout, i); in ChooseOperandLayoutFromOutputLayout() local
|
| D | dynamic_dimension_inference.cc | 1266 for (int64_t output_dim = 0; output_dim < output_rank; ++output_dim) { in HandleGather() local
|
| D | hlo_evaluator.cc | 1835 int64_t output_dim = in HandleGather() local
|
| D | elemental_ir_emitter.cc | 2158 int64 output_dim = operand_to_output_dim[operand_dim]; in EmitElementalGather() 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/ |
| D | optimizers_test.py | 34 def _get_model(input_dim, num_hidden, output_dim): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops.cc | 227 DimensionHandle output_dim; in CombinedNMSShapeFn() local
1026 DimensionHandle output_dim; in __anon4ac001e21b02() local
1053 DimensionHandle output_dim; in __anon4ac001e21c02() local
|
| /external/tensorflow/tensorflow/lite/kernels/gradient/ |
| D | bcast_grad_args.cc | 139 int output_dim = -1; in Invoke() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | bcast.h | 202 int output_dim = -1; in BCastList() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 500 if (auto output_dim = output_dimension_number(x_config[d])) { in Einsum() local
508 if (auto output_dim = output_dimension_number(y_config[d])) { in Einsum() local
|
| /external/XNNPACK/test/ |
| D | constant-pad-operator-tester.h | 84 inline size_t output_dim(size_t i) const { in output_dim() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2344 const int64_t output_dim = output_type.getDimSize(i); in Verify() local
|
| D | tf_ops_a_m.cc | 364 int64_t output_dim = output_shape[spatial_dim_index]; in Verify() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 3522 int64_t output_dim = 0; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
|