Searched refs:shape64 (Results 1 – 3 of 3) sorted by relevance
55 shape64 = constant_op.constant(shape, dtype=dtypes.int64)56 y = array_ops.reshape(x, shape64)
887 absl::Span<const int64_t> shape64 = pooling_descriptor.window(); in ScopedPoolingDescriptor() local897 std::transform(shape64.cbegin(), shape64.cend(), shape.begin(), in ScopedPoolingDescriptor()
941 absl::Span<const int64_t> shape64 = pooling_descriptor.window(); in CudnnPoolingDescriptor() local951 std::transform(shape64.cbegin(), shape64.cend(), shape.begin(), in CudnnPoolingDescriptor()