Home
last modified time | relevance | path

Searched refs:axis_index_ (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconcat_op.cc44 : XlaOpKernel(c), axis_index_(axis_index) {} in ConcatBaseOp()
47 const TensorShape concat_dim_tensor_shape = ctx->InputShape(axis_index_); in Compile()
54 ctx->ConstantInputAsIntScalar(axis_index_, &concat_dim)); in Compile()
97 int axis_index_; member in tensorflow::__anon8bb65f0c0111::ConcatBaseOp