| /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 | 42 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/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/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 220 for (int64_t output_dim : output_dims) { in FoldGatherOfGather() local
543 for (int64_t output_dim : operand->output_dims()) { in ReshapeToAddDegenerateDims() local
570 for (int64_t output_dim : new_output_dims) { in ReshapeToAddDegenerateDims() local
717 int64_t output_dim = scalar_indexed->output_dims()[i]; in FoldReshapeOfGatherNoDegenerateDims() local
881 auto is_broadcasted_dim = [&](int64_t output_dim) { in ComputeArrayForElementwiseBinaryOp()
|
| D | dynamic_padder.cc | 367 const int64_t output_dim = output_dims[i]; in GenerateBinaryMask() local
573 for (int64_t output_dim : output_dims) { in RewriteDynamicReshapeSplitInput() local
653 int64_t output_dim, absl::Span<HloInstruction*> input_dynamic_dims, in RewriteDynamicReshapeCombineInput()
795 int64_t output_dim = output_dims[0]; in RewriteDynamicReshapeSingleGroup() local
|
| D | sharding_propagation.cc | 140 int64_t output_dim = in IsConvolutionKernelSmall() local
|
| D | layout_assignment.cc | 1311 int64_t output_dim = LayoutUtil::Minor(output_layout, i); in ChooseOperandLayoutFromOutputLayout() local
|
| D | dynamic_dimension_inference.cc | 1404 for (int64_t output_dim = 0; output_dim < output_rank; ++output_dim) { in HandleGather() local
|
| D | elemental_ir_emitter.cc | 2165 int64_t output_dim = operand_to_output_dim[operand_dim]; in EmitElementalGather() local
|
| /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 | 108 output_dim, argument
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | argmax_spmd_expander.cc | 158 int output_dim = 0; in ComputeLayoutBackward() local
|
| D | reduce_spmd_expander.cc | 301 int output_dim = 0; in ComputeLayoutBackward() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | spacetobatch_op.cc | 81 const int64_t output_dim = in SpaceToBatch() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops.cc | 227 DimensionHandle output_dim; in CombinedNMSShapeFn() local
1026 DimensionHandle output_dim; in __anonf6f833bb1b02() local
1053 DimensionHandle output_dim; in __anonf6f833bb1c02() local
|
| D | nn_ops.cc | 1432 auto output_dim = c->MakeDim(output_dim_value); in ApproxTopKShape() local
|
| /external/tensorflow/tensorflow/lite/kernels/gradient/ |
| D | bcast_grad_args.cc | 140 int output_dim = -1; in Invoke() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| D | tf_jitrt_symbolic_shape_optimization.cc | 239 int64_t output_dim = out_type.getShape()[d]; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | replace_cast_hacks_with_tf_xla_ops.cc | 77 int output_dim, const SmallVector<int64_t> &weight_non_output_indices) { in CalculateZeroPointOffset()
|
| /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 | 83 inline size_t output_dim(size_t i) const { in output_dim() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 412 int64_t output_dim = output_shape[spatial_dim_index]; in verify() local
|