Searched refs:b_shape_tensor (Results 1 – 1 of 1) sorted by relevance
48 const auto& b_shape_tensor = ctx->input(8); in Compute() local55 b_shape_tensor, "b")); in Compute()62 ctx, a_shape_tensor.shape() == b_shape_tensor.shape(), in Compute()65 " vs b_shape: ", b_shape_tensor.shape().DebugString())); in Compute()82 const auto b_shape = b_shape_tensor.flat<int64_t>(); in Compute()