| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | kernelized_test.py | 195 def test_compute_output_shape(self, output_dim, initializer, scale): argument
223 def test_get_config(self, output_dim, initializer, scale, trainable): argument
253 def test_from_config(self, output_dim, initializer, scale, trainable): argument
281 def test_same_random_features_params_reused(self, output_dim, initializer, argument
|
| D | embeddings.py | 93 output_dim, argument
|
| D | kernelized.py | 152 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
|
| /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/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 | dynamic_padder.cc | 355 for (int64 output_dim : output_dims) { in RewriteDynamicReshapeSplitInput() local
379 const int64 output_dim = output_dims[i]; in RewriteDynamicReshapeSplitInput() local
472 for (int64 output_dim : output_dims) { in RewriteDynamicReshapeSplitInput() local
557 int64 output_dim, absl::Span<HloInstruction*> input_dynamic_dims, in RewriteDynamicReshapeCombineInput()
730 int64 output_dim = output_dims[0]; in RewriteDynamicReshapeSingleGroup() local
|
| D | sharding_propagation.cc | 136 int64 output_dim = in IsConvolutionKernelSmall() local
|
| D | layout_assignment.cc | 1098 int64 output_dim = LayoutUtil::Minor(output_layout, i); in ChooseOperandLayoutFromOutputLayout() local
|
| D | dynamic_dimension_inference.cc | 1228 for (int64 output_dim = 0; output_dim < output_rank; ++output_dim) { in HandleGather() local
|
| D | hlo_evaluator.cc | 1771 int64 output_dim = in HandleGather() local
|
| D | elemental_ir_emitter.cc | 1972 int64 output_dim = operand_to_output_dim[operand_dim]; in EmitElementalGather() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | optimizers_test.py | 38 def _get_model(input_dim, num_hidden, output_dim): argument
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | pad.cc | 83 int output_dim = output->dims->data[i]; in Prepare() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops.cc | 227 DimensionHandle output_dim; in CombinedNMSShapeFn() local
1026 DimensionHandle output_dim; in __anonc970e8b61b02() local
1053 DimensionHandle output_dim; in __anonc970e8b61c02() 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/core/util/ |
| D | bcast.h | 202 int output_dim = -1; in BCastList() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 467 if (auto output_dim = output_dimension_number(x_config[d])) { in Einsum() local
475 if (auto output_dim = output_dimension_number(y_config[d])) { in Einsum() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 3149 int64 output_dim = 0; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 364 int64_t output_dim = output_shape[spatial_dim_index]; in Verify() local
|
| D | tf_ops_n_z.cc | 2288 const int64_t output_dim = output_type.getDimSize(i); in Verify() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 5449 auto output_dim = output_tensor->getDimensions(); in ConvertFullyConnectedHelper() local
|