Home
last modified time | relevance | path

Searched defs:last_dim_size (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtopk_op.cc43 int last_dim_size = input_shape.dim_size(last_dim); in Compile() local
Ddiag_op.cc31 xla::XlaOp CreateDiagonal(xla::XlaOp input, int64 last_dim_size, in CreateDiagonal()
179 int64 last_dim_size = input_shape.dim_size(last_dim); in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Ddecode_proto_op.cc904 int last_dim_size; in AccumulateFields() member