Searched refs:axis_index_ (Results 1 – 1 of 1) sorted by relevance
44 : 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