Searched refs:perm_tensor_shape (Results 1 – 1 of 1) sorted by relevance
41 const TensorShape perm_tensor_shape = ctx->InputShape("perm"); in Compile() local44 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(perm_tensor_shape), in Compile()46 perm_tensor_shape.DebugString())); in Compile()49 OP_REQUIRES(ctx, dims == perm_tensor_shape.num_elements(), in Compile()53 perm_tensor_shape.num_elements())); in Compile()