| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | feature_column.py | 254 output_rank=2): argument
428 output_rank=2): argument
824 output_rank=2): argument
968 output_rank=2): argument
1197 output_rank=2): argument
1631 def _reshape_real_valued_tensor(input_tensor, output_rank, column_name=None): argument
1741 output_rank=2): argument
1881 output_rank=2): argument
2102 output_rank=2): argument
2344 output_rank=2): argument
[all …]
|
| D | feature_column_ops.py | 42 def _maybe_reshape_input_tensor(tensor, column_name, output_rank): argument
99 output_rank, argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | set_ops.cc | 53 DimensionHandle output_rank; in __anon967df2c80102() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 254 int64 output_rank) { in PermuteBatchAndOffsetDims()
316 int64 output_rank = output_shape.dimensions_size(); in ExpandInstruction() local
|
| D | instruction_fusion.cc | 173 int64 output_rank = 0; in EffectivelyAtMostUnary() local
|
| D | hlo_evaluator.cc | 783 int64 output_rank = output_shape.dimensions_size(); in IterationSpaceForOutputBatchIndices() local
800 int64 output_rank, absl::Span<const int64> slice_sizes, in IterationSpaceForOutputOffsetIndices()
|
| D | shape_inference.cc | 2631 const int64 output_rank = output_shape.rank(); in InferBroadcastShape() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util.cc | 51 const int64 output_rank = target_shape_in.NumElements(); in Reshape() local
|
| D | deserialize_sparse_variant_op.cc | 155 const int output_rank = output_shape->NumElements(); in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather_nd.cc | 84 const int output_rank = indices_rank + params_rank - indices_nd - 1; in Prepare() local
|
| D | embedding_lookup_sparse.cc | 154 const int output_rank = (lookup_rank - 1) + (embedding_rank - 1); in Eval() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 268 int64 output_rank = shape.rank(); in SourceIndexOfBroadcast() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 169 const int64 output_rank = output_config.size(); in Einsum() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1303 int64 output_rank = output_shape.rank(); in AlignLayouts() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 710 const int output_rank = in MaterializeReductionIndices() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1759 const int64 output_rank = output_shape.rank(); in Map() local
|