Searched refs:last_dim_size (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | topk_op.cc | 43 int last_dim_size = input_shape.dim_size(last_dim); in Compile() local
45 context, last_dim_size >= k, in Compile()
47 last_dim_size, ", needed ", k)); in Compile()
48 if (last_dim_size < k) { in Compile()
49 k = last_dim_size; in Compile()
|
| D | diag_op.cc | 31 xla::XlaOp CreateDiagonal(xla::XlaOp input, int64 last_dim_size, in CreateDiagonal() argument
44 xla::XlaOp iota = xla::Iota(builder, xla::S32, last_dim_size); in CreateDiagonal()
45 xla::XlaOp iota_broadcast = xla::Broadcast(iota, {last_dim_size}); in CreateDiagonal()
67 broadcast_dims.push_back(last_dim_size); in CreateDiagonal()
70 broadcast_dims[broadcast_dims.size() - 2] = last_dim_size; in CreateDiagonal()
179 int64 last_dim_size = input_shape.dim_size(last_dim); in Compile() local
184 xla::XlaOp diag = CreateDiagonal(input, last_dim_size, other_dims, in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | decode_proto_op.cc | 887 last_dim_size = tensor->dim_size(tensor->dims() - 1); in AccumulateFields()
892 stride = last_dim_size * element_size; in AccumulateFields()
898 stride = last_dim_size * sizeof(string); in AccumulateFields()
904 int last_dim_size; in AccumulateFields() member
928 default_value, info.last_dim_size); in AccumulateFields()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 2770 last_dim_size = array_ops.slice(
2772 output = array_ops.reshape(logits, array_ops.concat([[-1], last_dim_size], 0))
|